Commit d662c00e authored by galenxing's avatar galenxing
Browse files

pass per_task_metrics arg

parent e279904f

deepchem/models/tensorgraph/models/graph_models.py

+1 −1
Original line number Diff line number Diff line
@@ -614,7 +614,7 @@ class GraphConvTensorGraph(TensorGraph): 
        self.default_generator(dataset, predict=True),

        metrics,

        labels=self.my_labels,

        weights=[self.my_task_weights])

        weights=[self.my_task_weights], per_task_metrics = per_task_metrics)



  def bayesian_predict(self,

                       dataset,

CRA Git | Maintained and supported by SUSTech CRA and CCSE