Skip to content

Wrong type-casting for prediction of classification models #434

Open

Description

@valeriy42

When evaluating a multi-class classification model, it will predict a particular class name (e.g., a string). However, eland is trying to parse it as an integer (see here:

if self.model_type == TYPE_CLASSIFICATION:
            dt: "DTypeLike" = np.int_

This results in a failure, e.g., when I want to train and evaluate the iris classification dataset.

Activity

  1. added
    bugSomething isn't working
    topic:NLPIssue or PR about NLP model support and eland_import_hub_model
    on Feb 10, 2022
  2. added
    topic:mlIssue or PR about eland.ml.MLModel
    and removed
    topic:NLPIssue or PR about NLP model support and eland_import_hub_model
    on Oct 10, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingtopic:mlIssue or PR about eland.ml.MLModel

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions