Commit a60bc685 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Changes

parent 138522e7

deepchem/models/graph_models.py

+1 −1
Original line number Diff line number Diff line
@@ -265,7 +265,7 @@ class WeaveModel(KerasModel): 
      output_types = ['prediction', 'loss']

      loss: Loss = SoftmaxCrossEntropy()

    else:

      output = Dense(n_tasks)(weave_gather)

      output = Dense(n_tasks)(output)

      outputs = [output]

      output_types = ['prediction']

      loss = L2Loss()

CRA Git | Maintained and supported by SUSTech CRA and CCSE