search

geffy / tffm

TensorFlow implementation of an arbitrary order Factorization Machine
MIT License
780 stars 176 forks source link

Fix predict_proba bugs (python 2.7) #22

Closed xujin1982 closed 7 years ago

xujin1982 commented 7 years ago

I find the predict_proba in TFFMClassifier does not work correctly in Python 2.7. Just make a little change!