sklearn_demo's FFM always run error

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 518 forks source link

sklearn_demo's FFM always run error #314

Open ArcherZX opened 4 years ago

ArcherZX commented 4 years ago

The error appears on line 327, i think the value of 'num_features' is wrong The temp_file like this, Should the value of 'num_features' be '9992'?

ArcherZX commented 4 years ago

The value of 'num_features' in the program is 94914

txdas commented 4 years ago

you can fix it by this way fix