12cd0759c82480173978bbeeb8ef9ec5829e6542,deepchem/models/tests/test_generalize.py,TestGeneralization,test_sklearn_regression,#TestGeneralization#,34

Before Change

    transformers = []
    evaluator = Evaluator(model, test_dataset, transformers, verbosity=verbosity)
    scores = evaluator.compute_model_performance([regression_metric])
    print("scores")
    print(scores)

    assert scores[regression_metric.name] > .5

After Change

    verbosity = "high"
    regression_metric = Metric(metrics.r2_score, verbosity=verbosity)
    sklearn_model = LinearRegression()
    model = SklearnModel(sklearn_model, self.model_dir)

    // Fit trained model
    model.fit(train_dataset)
    model.save()

Instances