how to do unbalansed data classification, may you add weights as input parameter for example this has https://github.com/dstein64/PyFactorizationMachines