Commit c22d6fac authored by mufeili's avatar mufeili
Browse files

Update

parent f6abbc66

deepchem/models/torch_models/gcn.py

+212 −212
Original line number Diff line number Diff line
@@ -174,9 +174,9 @@ class GCN(nn.Module): 
      * When self.mode = 'regression',

        its shape will be ``(dgl_graph.batch_size, self.n_tasks)``.

      * When self.mode = 'classification', the output consists of probabilities

              for classes. Its shape will be

              ``(dgl_graph.batch_size, self.n_tasks, self.n_classes)`` if self.n_tasks > 1;

              its shape will be ``(dgl_graph.batch_size, self.n_classes)`` if self.n_tasks is 1.

        for classes. Its shape will be ``(dgl_graph.batch_size, self.n_tasks, self.n_classes)``

        if self.n_tasks > 1; its shape will be ``(dgl_graph.batch_size, self.n_classes)`` if

        self.n_tasks is 1.

      torch.Tensor, optional

        This is only returned when self.mode = 'classification', the output consists of the

        logits for classes before softmax.

deepchem/models/torch_models/gat.py

+204 −204

File changed.

Contains only whitespace changes.

deepchem/models/torch_models/attentivefp.py

+183 −183

File changed.

Contains only whitespace changes.

CRA Git | Maintained and supported by SUSTech CRA and CCSE