bug fixing (770cca00) · Commits · 钟慕尧 / deepchem

deepchem/molnet/run_benchmark.py

+9 −9

Original line number	Diff line number	Diff line
		@@ -156,9 +156,9 @@ def run_benchmark(datasets,
		n_features = list(train_dataset.get_data_shape())

		time_start_fitting = time.time()
		train_scores = {}
		valid_scores = {}
		test_scores = {}
		train_score = {}
		valid_score = {}
		test_score = {}

		if isinstance(model, str):
		if mode == 'classification':
		@@ -185,12 +185,12 @@ def run_benchmark(datasets,
		test=test)
		else:
		model.fit(train_dataset)
		train_scores['user_defined'] = model.evaluate(train_dataset, metric,
		train_score['user_defined'] = model.evaluate(train_dataset, metric,
		transformers)
		valid_scores['user_defined'] = model.evaluate(valid_dataset, metric,
		valid_score['user_defined'] = model.evaluate(valid_dataset, metric,
		transformers)
		if test:
		test_scores['user_defined'] = model.evaluate(test_dataset, metric,
		test_score['user_defined'] = model.evaluate(test_dataset, metric,
		transformers)

		time_finish_fitting = time.time()