search

gsganden / model_inspector

A uniform interface to a curated set of methods for inspecting machine learning models
https://gsganden.github.io/model_inspector/
Apache License 2.0
4 stars 0 forks source link

Warning about plotting on a grid #14

Open gsganden opened 2 years ago

gsganden commented 2 years ago

model_inspector's methods for plotting predictions for models with two features involve making predictions for a grid of possible feature values. Those grids are numpy arrays, so they do not have feature names. When the model is trained using a pandas DataFrame, the result is that those plotting methods give ugly warnings that are particularly distracting for beginners:

Screen Shot 2021-12-20 at 8 59 59 AM

I think the best solution is to turn the grids into DataFrames with column names from X.

gsganden commented 2 years ago

Another option is to suppress the warning