search

stevenpawley / Pyspatialml

Machine learning modelling for spatial data
GNU General Public License v3.0
145 stars 29 forks source link

Allow pandas pass through in predict method #44

Open stevenpawley opened 2 years ago

stevenpawley commented 2 years ago

Scikit learn can allow transformers to use column names instead of indexes, which is more convenient in some cases, e.g., ColumnTransformer. Currently, the use of names is not compatible with Pyspatialml because the data is read via numpy. Passing the data to the estimator via pandas will allow for preservation of column names