Commit eda15246 authored by leswing's avatar leswing
Browse files

Raise Learning Rate for IRV

parent f9b4834d

deepchem/models/tests/test_overfit.py

+1 −1
Original line number Diff line number Diff line
@@ -439,7 +439,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    classification_metric = dc.metrics.Metric(

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

    model = dc.models.TensorflowMultiTaskIRVClassifier(

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

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



    # Fit trained model

    model.fit(dataset_trans)

CRA Git | Maintained and supported by SUSTech CRA and CCSE