AnotherSamWilson / miceforest

Multiple Imputation with LightGBM in Python
MIT License
340 stars 31 forks source link

Compile predictions optimaly before impute_new_data is called #57

Closed AnotherSamWilson closed 2 months ago

AnotherSamWilson commented 2 years ago

If save_models == 1, procedure will be fastest and use least memory when predictions are compiled and deleted at the start/end of imputations for each dataset.

AnotherSamWilson commented 2 months ago

No longer necessary, predictions are automatically compiled unless save_all_iterations_data=False.