Organize the logic better in tox21_datasets.py (8cf240c6) · Commits · 钟慕尧 / deepchem

deepchem/molnet/load_function/tox21_datasets.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -12,13 +12,13 @@ import deepchem
		def load_tox21(featurizer='ECFP', split='index', reload=True, K=4):
		"""Load Tox21 datasets. Does not do train/test split"""
		# Featurize Tox21 dataset
		data_dir = deepchem.utils.get_data_dir()

		tox21_tasks = [
		'NR-AR', 'NR-AR-LBD', 'NR-AhR', 'NR-Aromatase', 'NR-ER', 'NR-ER-LBD',
		'NR-PPAR-gamma', 'SR-ARE', 'SR-ATAD5', 'SR-HSE', 'SR-MMP', 'SR-p53'
		]

		data_dir = deepchem.utils.get_data_dir()
		if reload:
		save_dir = os.path.join(data_dir, "tox21/" + featurizer + "/" + split)
		loaded, all_dataset, transformers = deepchem.utils.save.load_dataset_from_disk(