• Shining Sun's avatar
    Keras-fy NCF Model (#6092) · 048e5bff
    Shining Sun authored
    * tmp commit
    
    * tmp commit
    
    * first attempt (without eval)
    
    * Bug fixes
    
    * bug fixes
    
    * training done
    
    * Loss NAN, no eval
    
    * Loss weight problem solved
    
    * resolve the NAN loss problem
    
    * Problem solved. Clean up needed
    
    * Added a todo
    
    * Remove debug prints
    
    * Extract get_optimizer to ncf_common
    
    * Move metrics computation back to neumf; use DS.scope api
    
    * Extract DS.scope code to utils
    
    * lint fixes
    
    * Move obtaining DS above producer.start to avoid race condition
    
    * move pt 1
    
    * move pt 2
    
    * Update the run script
    
    * Wrap keras_model related code into functions
    
    * Update the doc for softmax_logitfy and change the method name
    
    * Resolve PR comments
    
    * working version with: eager, DS, batch and no masks
    
    * Remove git conflict indicator
    
    * move reshape to neumf_model
    
    * working version, not converge
    
    * converged
    
    * fix a test
    
    * more lint fix
    
    * more lint fix
    
    * more lint fixes
    
    * more lint fix
    
    * Removed unused imports
    
    * fix test
    
    * dummy commit for kicking of checks
    
    * fix lint issue
    
    * dummy input to kick off checks
    
    * dummy input to kick off checks
    
    * add collective to dist strat
    
    * addressed review comments
    
    * add a doc string
    048e5bff
distribution_utils.py 8.76 KB