Fix faulty LightGBM model
This commit is contained in:
parent
d9acdc9767
commit
2e1061af64
@ -104,9 +104,9 @@ class LightGBMPredictionModel(IFreqaiModel):
|
|||||||
|
|
||||||
# logger.info("--------------------Starting prediction--------------------")
|
# logger.info("--------------------Starting prediction--------------------")
|
||||||
|
|
||||||
original_feature_list = dk.find_features(unfiltered_dataframe)
|
dk.find_features(unfiltered_dataframe)
|
||||||
filtered_dataframe, _ = dk.filter_features(
|
filtered_dataframe, _ = dk.filter_features(
|
||||||
unfiltered_dataframe, original_feature_list, training_filter=False
|
unfiltered_dataframe, dk.training_features_list, training_filter=False
|
||||||
)
|
)
|
||||||
filtered_dataframe = dk.normalize_data_from_metadata(filtered_dataframe)
|
filtered_dataframe = dk.normalize_data_from_metadata(filtered_dataframe)
|
||||||
dk.data_dictionary["prediction_features"] = filtered_dataframe
|
dk.data_dictionary["prediction_features"] = filtered_dataframe
|
||||||
|
Loading…
Reference in New Issue
Block a user