Hi there,
I recently started using PyTorch (more familiar with Keras) and here is 1st my attempt at Factorization Machine:

  
      
      GitHub
  
  
    

mzaradzki/factorization-machine-for-prediction 135

Factorization Machine for regression and classification - mzaradzki/factorization-machine-for-prediction


  
  
    
    
  
  


Remark ; I implemented it using the O(k.N) formulation found in Steffen Rendle paper