columns get renamed when `predict` is called

mljar / mljar-api-python

A simple python wrapper over MLJAR API.

https://docs.mljar.com/

Apache License 2.0

42 stars 10 forks source link

columns get renamed when `predict` is called #5

Closed shadiakiki1986 closed 6 years ago

shadiakiki1986 commented 6 years ago

Hi. I used the predict method on a pandas dataframe with the same column names of the dataframe used in the fit method, but the columns with the predict method got renamed from 1 to attribute_1, 2 to attribute_2, etc. Is this because the column names are numeric?

pplonski commented 6 years ago

You are right - this is because the column names are numeric. If you want to keep your column name, please use other column name, for example feature1 .