12cd0759c82480173978bbeeb8ef9ec5829e6542,deepchem/models/tests/test_generalize.py,TestGeneralization,test_sklearn_transformed_regression,#TestGeneralization#,83

Before Change

        transformer.transform(test_dataset)

    tasks = train_dataset.get_task_names()
    task_types = {task: "regression" for task in tasks}

    model_params = {
      "batch_size": None,

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