Skip to content

Corner Case - One input feature and prediction #131

@ChandraLingam

Description

@ChandraLingam

There is an issue with csv serializer when running prediction on a model that has only one input feature. Serializer or the SageMaker Prediction endpoint is not able to handle single feature models when running predictions. I used CSV for inference format.

I don't have the exact error message that was reported by the serializer; wanted to report this issue.

I trained the data with XGBoost algorithm

Data set consisted of just X, y
X, y
1,1
2,4
3,9
....

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions