xlearn icon indicating copy to clipboard operation
xlearn copied to clipboard

How to set class weight for target label.

Open sandiemann opened this issue 5 years ago • 3 comments

Hello,

I am dealing with highly imbalance data and I was wondering if there is an option to set the class weight for the target labels for FFM model training to basically under-sample the negative class.

Regards, Sanchit

sandiemann avatar Aug 08 '18 12:08 sandiemann

I have the same problem for FM model. Is is possible to tweak the source code for setting class weight for the minority class?

popeye007 avatar Aug 14 '18 04:08 popeye007

@sandiemann @popeye007 I will add this feature to xLearn in the next version. Thanks so much!

aksnzhy avatar Sep 14 '18 08:09 aksnzhy

@aksnzhy any updates on this? Is it documented? Thank you.

thestick613 avatar Jan 20 '20 08:01 thestick613