Minor fix with HIV dataset (7cd735c0) · Commits · 钟慕尧 / deepchem

deepchem/molnet/load_function/hiv_datasets.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -28,6 +28,8 @@ def load_hiv(featurizer='ECFP',
		if save_dir is None:
		save_dir = DEFAULT_DIR

		hiv_tasks = ["HIV_active"]

		save_folder = os.path.join(save_dir, "hiv-featurized", str(featurizer),
		str(split))
		if featurizer == "smiles2img":
		@@ -44,8 +46,6 @@ def load_hiv(featurizer='ECFP',
		if not os.path.exists(dataset_file):
		deepchem.utils.download_url(url=HIV_URL, dest_dir=data_dir)

		hiv_tasks = ["HIV_active"]

		if featurizer == 'ECFP':
		featurizer = deepchem.feat.CircularFingerprint(size=1024)
		elif featurizer == 'GraphConv':