automl/Auto-PyTorch

Prediction on one sample produces error

AlexanderZender opened this issue · 1 comments

AlexanderZender commented

NOTE: ISSUES ARE NOT FOR CODE HELP - Ask for Help at https://stackoverflow.com

Your issue may already be reported!
Also, please search on the issue tracker before creating one.

  • I'm submitting a ...
    • bug report
    • feature request
    • support request => Please do not submit support request here, see note at the top of this template.

Issue Description

  • When calling the proba_predict method with a single sample the following error is produced:

image
Prediction sets with more than one work just fine.

Expected Behavior

A single probability is returned.

Current Behavior

This error is returned deep within the predict pipeline:
image

Possible Solution

Your Code

image

Error message

image

Your Local environment

  • Operating System, Ubuntu
  • Python, version 3.10
  • Outputs of pip freeze or conda list
AlexanderZender commented

My current workaround, duplicating the single row and only use the probabilities from one:
image