Formatting: (ce95dbcf) · Commits · 钟慕尧 / deepchem

devtools/jenkins/compare_results.py

+3 −2

Original line number	Diff line number	Diff line
		@@ -61,8 +61,9 @@ def is_good_result(my_result, desired_result):
		# Higher is Better
		desired_value = desired_result[key] * (1.0 - CUSHION_PERCENT)
		if my_result[key] < desired_value:
		message_part = "%s,%s,%s,%s,%s,%s" % (my_result['data_set'], my_result['model'], my_result['split'],
		key, my_result[key], desired_result[key])
		message_part = "%s,%s,%s,%s,%s,%s" % (
		my_result['data_set'], my_result['model'], my_result['split'], key,
		my_result[key], desired_result[key])
		message.append(message_part)
		retval = False
		return retval, message

+4 −1

Original line number	Diff line number	Diff line
		@@ -6,7 +6,9 @@ import sys

		def classification_table_to_csv(lines):
		output = []
		headers = ["split", "dataset", "model", "Train score/ROC-AUC", "Valid score/ROC-AUC"]
		headers = [
		"split", "dataset", "model", "Train score/ROC-AUC", "Valid score/ROC-AUC"
		]
		output.append(",".join(headers))
		for line in lines:
		vars = [x.strip() for x in line.split('\|')]
		@@ -28,6 +30,7 @@ def classification_table_to_csv(lines):
		for l in output:
		print(l)


		def regression_table_to_csv(lines):
		output = []

+6 −3

Original line number	Diff line number	Diff line
		@@ -55,9 +55,12 @@ def gen_kaggle(KAGGLE_tasks,
		############################################################## TIMING
		# Set some global variables up top
		current_dir = os.path.dirname(os.path.realpath(__file__))
		train_files = os.path.join(current_dir, "KAGGLE_training_disguised_combined_full.csv.gz")
		valid_files = os.path.join(current_dir, "KAGGLE_test1_disguised_combined_full.csv.gz")
		test_files = os.path.join(current_dir, "KAGGLE_test2_disguised_combined_full.csv.gz")
		train_files = os.path.join(current_dir,
		"KAGGLE_training_disguised_combined_full.csv.gz")
		valid_files = os.path.join(current_dir,
		"KAGGLE_test1_disguised_combined_full.csv.gz")
		test_files = os.path.join(current_dir,
		"KAGGLE_test2_disguised_combined_full.csv.gz")

		# Featurize KAGGLE dataset
		print("About to featurize KAGGLE dataset.")