Commit ae4f1f5a authored by Michelle Gill's avatar Michelle Gill
Browse files

Fix GraphConvTensorGraph to GraphConvModel in qm7 example

parent 8090fa9c
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -25,7 +25,7 @@ metric = [
# Batch size of models
batch_size = 64

model = dc.models.GraphConvTensorGraph(
model = dc.models.GraphConvModel(
    len(tasks), batch_size=batch_size, learning_rate=0.001, mode="regression")

# Fit trained model