Turn it into a prediction (f69999e1) · Commits · 钟慕尧 / deepchem

contrib/atomicconv/acnn/core/opt_scaffold.py

+5 −5

Original line number	Diff line number	Diff line
		@@ -85,7 +85,7 @@ train_evaluator = dc.utils.evaluate.Evaluator(model, train_dataset,
		train_scores = train_evaluator.compute_model_performance(metric)
		print("Train scores")
		print(train_scores)
		test_evaluator = dc.utils.evaluate.Evaluator(model, test_dataset, transformers)
		test_scores = test_evaluator.compute_model_performance(metric)
		print("Test scores")
		print(test_scores)

		print("Prediction")
		predictions = model.predict(test_dataset, transformers)
		print(predictions)