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

Rename GraphConvTensorGraph to GraphConvModel in actual class

parent 66628f7b
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -664,7 +664,7 @@ class PetroskiSuchTensorGraph(TensorGraph):
        per_task_metrics=per_task_metrics)


class GraphConvTensorGraph(TensorGraph):
class GraphConvModel(TensorGraph):

  def __init__(self,
               n_tasks,
@@ -700,7 +700,7 @@ class GraphConvTensorGraph(TensorGraph):
    self.graph_conv_layers = graph_conv_layers
    kwargs['use_queue'] = False
    self.number_atom_features = number_atom_features
    super(GraphConvTensorGraph, self).__init__(**kwargs)
    super(GraphConvModel, self).__init__(**kwargs)
    self.build_graph()

  def build_graph(self):
@@ -824,7 +824,7 @@ class GraphConvTensorGraph(TensorGraph):
          result = undo_transforms(feed_results[0], transformers)
          feed_results = [result]
        for ind, result in enumerate(feed_results):
          # GraphConvTensorGraph constantly outputs batch_size number of
          # GraphConvModel constantly outputs batch_size number of
          # results, only valid samples should be appended to final results
          results[ind].append(result[:n_samples])