def test_binary_classifier_from_loaded_model(self):
model_path = get_temp_model_file()
self.binary_model.save_model(model_path)
loaded_model = Pipeline()
loaded_model.load_model(model_path)
pfi_from_loaded = loaded_model.permutation_feature_importance(
self.classification_data)
assert_frame_equal(self.binary_pfi, pfi_from_loaded)
os.remove(model_path)
def test_ranker_from_loaded_model(self):
model_path = get_temp_model_file()
self.ranker_model.save_model(model_path)
loaded_model = Pipeline()
loaded_model.load_model(model_path)
pfi_from_loaded = loaded_model.permutation_feature_importance(
self.ranking_data)
assert_frame_equal(self.ranker_pfi, pfi_from_loaded)
os.remove(model_path)
