Commit 9f6a0048 authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

small fix

parent 1462afad

examples/benchmark.py

+1 −1
Original line number Diff line number Diff line
@@ -535,7 +535,7 @@ if __name__ == '__main__': 
  hps['logreg'] = [{'penalty': 0.1, 'penalty_type': 'l2', 'batch_size': 50, 

                    'nb_epoch': 10, 'learning_rate': 0.005}]

                

  hps['graphconv'] = [{'batch_size': 50, 'nb_epoch': 10, 

  hps['graphconv'] = [{'batch_size': 50, 'nb_epoch': 15, 

                       'learning_rate': 0.0005, 'n_filters': 64, 

                       'n_fully_connected_nodes': 128, 'seed': 123}]

examples/delaney/delaney_graph_conv.py

+1 −1
Original line number Diff line number Diff line
@@ -29,7 +29,7 @@ with g.as_default(): 


  # Do setup required for tf/keras models

  # Number of features on conv-mols

  n_feat = 71

  n_feat = 75

  # Batch size of models

  batch_size = 128

  graph_model = dc.nn.SequentialGraph(n_feat)

examples/tox21/tox21_graph_conv.py

+1 −1
Original line number Diff line number Diff line
@@ -30,7 +30,7 @@ with g.as_default(): 


  # Do setup required for tf/keras models

  # Number of features on conv-mols

  n_feat = 71

  n_feat = 75

  # Batch size of models

  batch_size = 50

  graph_model = dc.nn.SequentialGraph(n_feat)

CRA Git | Maintained and supported by SUSTech CRA and CCSE