PetroskiSuchTensorGraph to PetroskiSuchModel

    return out

class PetroskiSuchTensorGraph(TensorGraph):

class PetroskiSuchModel(TensorGraph):

  """

      Model from Robust Spatial Filtering with Graph Convolutional Neural Networks

      https://arxiv.org/abs/1703.00792

    self.error_bars = True if 'error_bars' in kwargs and kwargs['error_bars'] else False

    self.dropout = dropout

    kwargs['use_queue'] = False

    super(PetroskiSuchTensorGraph, self).__init__(**kwargs)

    super(PetroskiSuchModel, self).__init__(**kwargs)

    self.build_graph()

  def build_graph(self):

tf.set_random_seed(123)

import deepchem as dc

from deepchem.molnet import load_tox21

from deepchem.models.tensorgraph.models.graph_models import PetroskiSuchTensorGraph

from deepchem.models.tensorgraph.models.graph_models import PetroskiSuchModel

model_dir = "/tmp/graph_conv"

# Batch size of models

batch_size = 128

model = PetroskiSuchTensorGraph(

model = PetroskiSuchModel(

    len(tox21_tasks), batch_size=batch_size, mode='classification')

model.fit(train_dataset, nb_epoch=10)