IDA-HumanCapital/fife

[Feature request] multiindex for forecasts from exitmodeler

Closed this issue · 0 comments

LGBExitModeler().forecast() currently outputs the exit type as the last column. It would be more intuitive to use a multi-index structure that contains the individual identifier and the exit type. This would ensure the entire DataFrame is of the same dtype, which would make manipulations easier.