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

changes

parent 7d1f4c85

deepchem/models/tests/test_overfit.py

+1 −1
Original line number Diff line number Diff line
@@ -429,7 +429,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    dataset_trans = IRV_transformer.transform(dataset)

    classification_metric = dc.metrics.Metric(

        dc.metrics.accuracy_score, task_averager=np.mean)

    model = dc.models.TensorflowMultitaskIRVClassifier(

    model = dc.models.MultitaskIRVClassifier(

        n_tasks, K=5, learning_rate=0.01, batch_size=n_samples)



    # Fit trained model

CRA Git | Maintained and supported by SUSTech CRA and CCSE