Commit cfc588a1 authored by miaecle's avatar miaecle
Browse files

update fix

parent b245c596

deepchem/models/tensorgraph/models/graph_models.py

+1 −1
Original line number Diff line number Diff line
@@ -619,7 +619,7 @@ class GraphConvTensorGraph(TensorGraph): 
            self.layers[k.name].out_tensor: v

            for k, v in six.iteritems(feed_dict)

        }

        n_samples = feed_dict[self.labels[0].out_tensor].shape[0]

        n_samples = max(feed_dict[self.membership.out_tensor]) + 1



        feed_dict[self._training_placeholder] = 0.0

        feed_results = self.session.run(outputs, feed_dict=feed_dict)

CRA Git | Maintained and supported by SUSTech CRA and CCSE