AxeldeRomblay / MLBox

MLBox is a powerful Automated Machine Learning python library.
https://mlbox.readthedocs.io/en/latest/
Other
1.49k stars 274 forks source link

dict target #34

Closed ilyes495 closed 7 years ago

ilyes495 commented 7 years ago

I have a question regarding the format of target dataframe which is passed to fit_predict(), does it contain the labels of training dataset or for test dataset as well ? and how it should be formatted? I think it's not clear if someone has to process the data himself before passing it to the predictor.

Thanks

AxeldeRomblay commented 7 years ago

The "target dataframe" (actually it must be a pandas Serie) must contain the labels of training dataset only.

Also if you're dealing with a classification problem, you need to have the target encoder python object in the directory "save" (LabelEncoderfitted) and saved as "target_encoder.obj"

AxeldeRomblay commented 7 years ago

I will add more details to the docs tomorrow ! Thanks for reporting this issue !