save and load is unnecessarily complicated
I think it would be much simpler to just pickle the model instead of selecting the attributes and saving stuff as a .npz file...
I think it would be much simpler to just pickle the model instead of selecting the attributes and saving stuff as a .npz file...