xlearn icon indicating copy to clipboard operation
xlearn copied to clipboard

only binary classification ??

Open ucasiggcas opened this issue 4 years ago • 1 comments

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

ucasiggcas avatar Jun 27 '20 04:06 ucasiggcas

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

Chenlonw avatar Nov 10 '20 03:11 Chenlonw