feature importance plot (lollipop chart for example)

UBC-MDS / DSCI_522_group_401

Medical Expenses Prediction

MIT License

0 stars 4 forks source link

Closed sreejithmunthikodu closed 4 years ago

sreejithmunthikodu commented 4 years ago

get feature importance and make a plot

sreejithmunthikodu commented 4 years ago

Since we have categorical features which are one-hot-encoded and polynomial features of degree 2, it is hard to interpret a feature_importance plot. Hence we are not implementing it in this work. We may fit a model without the polynomial features for interpreting feature importance. However, since we are more interested in prediction and also considering the limited time available, we are not including it in this work.