Commit 8090fa9c authored by Michelle Gill's avatar Michelle Gill
Browse files

Fix GraphConvTensorGraph to GraphConvModel in hopv example

parent 63629e07
Loading
Loading
Loading
Loading
+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
@@ -30,7 +30,7 @@ metric = [
n_feat = 75
# Batch size of models
batch_size = 50
model = GraphConvTensorGraph(
model = GraphConvModel(
    len(hopv_tasks), batch_size=batch_size, mode='regression')

# Fit trained model