Commit 74d47a73 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #735 from galenxing/master 

Pass per_task_metrics arg to evaluate_generator()
parents 7972a62c 923103c1

deepchem/models/tensorgraph/models/graph_models.py

+2 −1
Original line number Diff line number Diff line
@@ -616,7 +616,8 @@ 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