Commit 8b318e90 authored by abster12's avatar abster12
Browse files

test for pickle added

parent 47bec3c3

deepchem/models/tensorgraph/tests/test_layers_pickle.py

+10 −0
Original line number Diff line number Diff line
@@ -682,3 +682,13 @@ def test_Slice_pickle(): 
  tg.set_loss(out)

  tg.build()

  tg.save()





def test_hingeloss_pickle():

  tg = TensorGraph()

  feature = Feature(shape=(tg.batch_size, 1))

  layer = Hingeloss(in_layers=[feature, feature])

  tg.add_output(layer)

  tg.set_loss(layer)

  tg.build()

  tg.save()

CRA Git | Maintained and supported by SUSTech CRA and CCSE