def test_predict_output_2d(self):
"""Test prediction."""
areg = AdvancedTransformedTargetRegressor(
transformer=StandardScaler(),
regressor=self.Reg2DPrediction(),
)
areg.fit(np.arange(3).reshape(3, 1), np.arange(3))
pred = areg.predict([[1]])
np.testing.assert_allclose(areg.transformer_.scale_, [0.8164965809])
np.testing.assert_allclose(areg.transformer_.mean_, [1.0])
np.testing.assert_allclose(pred, [42.0 * 0.8164965809 + 1.0])
def test_feature_importances_(self):
"""Test ``feature_importances_`` property."""
areg = AdvancedTransformedTargetRegressor(
regressor=FeatureImportanceRegressor())
areg.fit(np.arange(3).reshape(3, 1), np.arange(3))
assert areg.feature_importances_ == 42
def test_coef_(self):
"""Test ``coef_`` property."""
areg = AdvancedTransformedTargetRegressor()
areg.fit(np.arange(3).reshape(3, 1), np.arange(3))
np.testing.assert_allclose(areg.coef_, [1.0])
def test_regressor_none(self):
"""Test ``regressor=None``."""
areg = AdvancedTransformedTargetRegressor()
areg.fit(np.arange(3).reshape(3, 1), np.arange(3))
assert isinstance(areg.regressor_, LinearRegression)