Commit 63629e07 authored by Michelle Gill's avatar Michelle Gill
Browse files

Fix GraphConvTensorGraph to GraphConvModel in delaney example

parent b98df28c

examples/delaney/delaney_graph_conv.py

+2 −2
Original line number Diff line number Diff line
@@ -7,7 +7,7 @@ from __future__ import unicode_literals 


import numpy as np



from models import GraphConvTensorGraph

from models import GraphConvModel



np.random.seed(123)

import tensorflow as tf
@@ -29,7 +29,7 @@ metric = dc.metrics.Metric(dc.metrics.pearson_r2_score, np.mean) 
n_feat = 75

# Batch size of models

batch_size = 128

model = GraphConvTensorGraph(

model = GraphConvModel(

    len(delaney_tasks), batch_size=batch_size, mode='regression')



# Fit trained model

examples/delaney/delaney_graphconv_error_bars.py

+1 −1
Original line number Diff line number Diff line
@@ -27,7 +27,7 @@ delaney_tasks, delaney_datasets, transformers = dc.molnet.load_delaney( 
train_dataset, valid_dataset, test_dataset = delaney_datasets

metric = dc.metrics.Metric(dc.metrics.pearson_r2_score, np.mean)



model = dc.models.GraphConvTensorGraph(

model = dc.models.GraphConvModel(

    len(delaney_tasks),

    batch_size=BATCH_SIZE,

    learning_rate=LR,

CRA Git | Maintained and supported by SUSTech CRA and CCSE