search

jimthompson5802 / model-stacking-workbench

Framework for model stacking that can be applied to Kaggle competitions
MIT License
3 stars 1 forks source link

ModelTrainer - predict function not correctly formating data structure #3

Closed jimthompson5802 closed 6 years ago

jimthompson5802 commented 6 years ago

Currently, the ModelTrainer for regression prediction creates np.array of shape (#rows,). This causes errors when writing the predicitons for the test data set. Need to reshape predictions from (#row,) to (#row,1).

jimthompson5802 commented 6 years ago

Added .reshape(-1,1) to all regression predict sha1: f7f89e17a3ac773e7dfac62232f5fbbff69e50a0