def test_blending():
assert_submission(ramp_kit_dir=os.path.join(PATH, "kits", "iris"),
ramp_data_dir=os.path.join(PATH, "kits", "iris"),
ramp_submission_dir=os.path.join(PATH, "kits", "iris",
"submissions"),
submission='starting_kit',
is_pickle=True,
save_output=True,
retrain=True)
assert_submission(ramp_kit_dir=os.path.join(PATH, "kits", "iris"),
ramp_data_dir=os.path.join(PATH, "kits", "iris"),
ramp_submission_dir=os.path.join(PATH, "kits", "iris",
"submissions"),
submission='random_forest_10_10',
is_pickle=True,
save_output=True,
retrain=True)
blend_submissions(['starting_kit', 'random_forest_10_10'],
ramp_kit_dir=os.path.join(PATH, "kits", "iris"),
ramp_data_dir=os.path.join(PATH, "kits", "iris"),
ramp_submission_dir=os.path.join(PATH, "kits", "iris",
"submissions"),
save_output=True)
# cleaning up so next test doesn't try to train "training_output"
shutil.rmtree(
os.path.join(PATH, "kits", "iris", "submissions", "training_output"))
def test_blending():
assert_submission(ramp_kit_dir=os.path.join(PATH, "kits", "iris"),
ramp_data_dir=os.path.join(PATH, "kits", "iris"),
submission='starting_kit',
is_pickle=True,
save_y_preds=True,
retrain=True)
assert_submission(ramp_kit_dir=os.path.join(PATH, "kits", "iris"),
ramp_data_dir=os.path.join(PATH, "kits", "iris"),
submission='random_forest_10_10',
is_pickle=True,
save_y_preds=True,
retrain=True)
blend_submissions(['starting_kit', 'random_forest_10_10'],
ramp_kit_dir=os.path.join(PATH, "kits", "iris"),
ramp_data_dir=os.path.join(PATH, "kits", "iris"))