Commit 2d1e038e authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

fix for failure

parent 259a4fd2
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -483,7 +483,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
    classification_metric = dc.metrics.Metric(
      dc.metrics.accuracy_score, verbosity=verbosity, task_averager=np.mean)
    tensorflow_model = dc.models.TensorflowLogisticRegression(
        n_tasks, n_features, learning_rate=0.5 weight_init_stddevs=[.01],
        n_tasks, n_features, learning_rate=0.5, weight_init_stddevs=[.01],
        batch_size=n_samples, verbosity=verbosity)
    model = dc.models.TensorflowModel(tensorflow_model)