xlearn
xlearn copied to clipboard
only binary classification ??
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
https://github.com/aksnzhy/xlearn/issues/197