Skip to content
forked from rpasricha/TransFM

TF 2.x implementation of PRME, with features to optimize for commercial use

Notifications You must be signed in to change notification settings

macfarll/TransFM

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is a work-in-progress rework of the TransFM code here: https://github.com/rpasricha/TransFM

For a high level demonstration of the code see here: https://colab.research.google.com/drive/1oAq1cGgo7XttF0Zpgr55zF8gtT4VdNEE?usp=sharing

More feature demos and documentation will be added soon.

While the original code was from the above repository, the changes here are extensive and include:

-Changing Tensorflow version to 2.x

-Adding functionality for outputting predections

-Memory Optimizations most notably the ability to train in segments

-Adding option to use regional data, limiting users to interacting with only certain items

-Ability to save and load model in a way that preserves user data

-Ability to add new items and users to a partially trained model

-Option to only use train/test, allowing more users to be covered by model

-Model seeding allows for determininsitic results

-Args object allows for easier debugging and saving of run parameters

-Added pyton notebook that can be used in google colab to quickly demo model

About

TF 2.x implementation of PRME, with features to optimize for commercial use

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 60.9%
  • Python 39.1%