Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Predicting Probability #98

Open
econeer opened this issue May 6, 2017 · 2 comments
Open

Predicting Probability #98

econeer opened this issue May 6, 2017 · 2 comments
Labels

Comments

@econeer
Copy link

econeer commented May 6, 2017

Probably fastFM is more for the classification but do we have a way to predict probability using fastFM? Although fastFM has predict_proba() function, it seems like it's not an actual probability (more like a score it seems). For example, I had data (over 100,000) which has 30% True events and 70% False events. Therefore, I expected average output of predict_prob() to be 30% (which I get from Logistic Regression) but fastFM generated average probability of 43%. Are there parameters I should change?

@ibayer
Copy link
Owner

ibayer commented May 6, 2017

The Logistic Regression is calibrated by default which is not the case for many model. fastFM uses probit regression (with als solver) for classification. However, the probabilities can be calibrated see http://scikit-learn.org/stable/modules/calibration.html for more details.

@econeer
Copy link
Author

econeer commented May 6, 2017

I'll check it out. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

2 participants