Implementation of regression model - Githubissues

geffy / tffm

TensorFlow implementation of an arbitrary order Factorization Machine

MIT License

780 stars 176 forks source link

Implementation of regression model #6

Closed geffy closed 8 years ago

geffy commented 8 years ago

Refactored code:

customizable loss function in core graph
abstract class TFFMBaseModel
separate classes for classification and regression