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.08k stars 518 forks source link

only binary classification ？？ #331

Open ucasiggcas opened 4 years ago

ucasiggcas commented 4 years ago

hi,dear as a poor learner for FM/FFM, wanna know the classification only has 2 classes ?? also see the demo multi-classes is where ? and how to set the param in FMModel?

FMModel(model_type='fm', task='binary', metric='auc', block_size=500, lr=0.2, k=4, reg_lambda=0.1, init=0.1, f
old=1, epoch=5, stop_window=2, opt='sgd', nthread=None, n_jobs=4, alpha=1, beta=1, lambda_1=1, lambda_2=1, **kwarg
s)

btw,the metric is only one ??could be more ? that is,such as metric=['auc','f1','prec'] ?tried, but not success,

could you pls help me? thx

Chenlonw commented 3 years ago

https://github.com/aksnzhy/xlearn/issues/197