--

store the original data frame in another df and use that. column() only works with data frames...

orig_train_data = dfTrain.drop('Outcome',1)

and then in this section use ...

labels = list(orig_train_data.columns)

--

--

Venki Ramachandran
Venki Ramachandran

Written by Venki Ramachandran

Articles on setting up and managing Global Capability Centers. Technical articles on AWS, Cloud computing, and developing applications for scale and performance

No responses yet