Lm predict forward no labels in matrix

dssg / triage

General Purpose Risk Modeling and Prediction Toolkit for Policy and Social Good Problems

Other

181 stars 61 forks source link

Lm predict forward no labels in matrix #942

Closed silil closed 1 month ago

silil commented 3 months ago

Modify the builders.py component to deal with the use case of not having labels on the matrix (predict forward). In case there are no labels, a CSV file is generated with 0s in it. This change follows the logic on tag 5.2.3.

It was tested with DoJo data using the predictlist logic:

Using only the predict_forward_with_existed_model
Using the Retrainer object with retrain and predict methods

KasunAmare commented 1 month ago

Changes to the builders.py look reasonable to me. Also, tested this branch on the acdhs_housing predict forward script, and it works.