def test_folding_regressor_functions():
"""Testing folding functions """
data, y, sample_weight = generate_classification_data()
for X in [data, numpy.array(data)]:
kfolder = FoldingRegressor(SklearnRegressor(GradientBoostingRegressor(n_estimators=5)), n_folds=2)
kfolder.fit(X, y, sample_weight=sample_weight)
preds = kfolder.predict(X)
for p in kfolder.staged_predict(X):
pass
assert numpy.allclose(p, preds)
importances = kfolder.feature_importances_
other_importances = kfolder.get_feature_importances()
def test_folding_regressor_functions():
"""Testing folding functions """
data, y, sample_weight = generate_classification_data()
for X in [data, numpy.array(data)]:
kfolder = FoldingRegressor(SklearnRegressor(GradientBoostingRegressor(n_estimators=5)), n_folds=2)
kfolder.fit(X, y, sample_weight=sample_weight)
preds = kfolder.predict(X)
for p in kfolder.staged_predict(X):
pass
assert numpy.allclose(p, preds)
importances = kfolder.feature_importances_
other_importances = kfolder.get_feature_importances()
