Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Add float32 support for SGDClassifier and SGDRegressor #5776

@elanmart

Description

@elanmart

Hi,

I was wondreing why sklearn does not allow me to specify the dtype I'd like to use.

I'm working with rather large dataset, which fits into my RAM as float32, but when I'm trying to train a simple SGD on it, the model tries to copy my data into float64, causing MemoryError.

I can change this in my local sklearn build, but I guess there is a good reason why this is not a free parameter?

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions