search

aksnzhy / xlearn

High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.
https://xlearn-doc.readthedocs.io/en/latest/index.html
Apache License 2.0
3.09k stars 519 forks source link

setting batch_size #260

Closed yanghzcc closed 5 years ago

yanghzcc commented 5 years ago

I can see "int batch_size = 1000000;" in source, but how can i set the parameter of batch_size in python interface? I add 'batch_size ' in para, and get a error

aksnzhy commented 5 years ago

@yanghzcc Sorry, xLearn uses SGD and you cannot set the batch_size parameter.

yanghzcc commented 5 years ago

thanks @aksnzhy