Commit 716385f0 authored by cc's avatar cc
Browse files

Fixed graphconv

parent 14855aa7

examples/chembl/chembl_graph_conv.py

+1 −1
Original line number Diff line number Diff line
@@ -21,7 +21,7 @@ K.set_session(sess) 
with g.as_default():

  tf.set_random_seed(123)

  chembl_tasks, datasets, transformers = load_chembl(shard_size=2000,

    featurizer="ECFP", set="5thresh", split="random")

    featurizer="GraphConv", set="5thresh", split="random")

  train_dataset, valid_dataset, test_dataset = datasets



  # Fit models

CRA Git | Maintained and supported by SUSTech CRA and CCSE