Fixing chembl example (76ed80e4) · Commits · 钟慕尧 / deepchem

docs/examples.rst

+76 −77

Original line number	Diff line number	Diff line
		@@ -103,7 +103,7 @@ For a :class:`GraphConvModel <deepchem.models.GraphConvModel>`, we'll reload our
		>>> assert valid_scores['mean-pearson_r2_score'] > 0.3, valid_scores


		..

		ChEMBL
		-------

		@@ -134,15 +134,14 @@ For a :class:`GraphConvModel <deepchem.models.GraphConvModel>`, we'll reload our
		>>> n_layers = 3
		>>> model = dc.models.MultitaskRegressor(
		... len(chembl_tasks),
		... train_dataset.get_data_shape()[0],
		... n_features=1024,
		... layer_sizes=[1000] * n_layers,
		... dropouts=[.25] * n_layers,
		... weight_init_stddevs=[.02] * n_layers,
		... bias_init_consts=[1.] * n_layers,
		... learning_rate=.0003,
		... weight_decay_penalty=.0001,
		... batch_size=100,
		... verbosity="high")
		... batch_size=100)
		>>>
		>>> model.fit(train_dataset, nb_epoch=20)
		0...