Skip to content

[BUG] Fix the scikit-learn miscellaneous/plot_set_output.py example #6498

Open
@csadorf

Description

@csadorf

The scikit-learn miscellaneous/plot_set_output.py example fails when running with the cuML accelerator due to a dimension mismatch between the logistic regression coefficients and feature names.

Error Message

ValueError: Length of values (5) does not match length of index (981)

Steps to Reproduce

Run the example with cuML accelerator:

python -m cuml.accel miscellaneous/plot_set_output.py

Expected Behavior

The example should run successfully.

Environment

  • Python 3.12.9
  • cuML 25.04 (nightlies)
  • scikit-learn 1.5.2

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingcuml-accelIssues related to cuml.accel

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions