• Shining Sun's avatar
    Added benchmark test and convergence test for the NCF model (#6318) · 4c11b84b
    Shining Sun authored
    * initial commit
    
    * bug fix
    
    * Move build_stats from common to keras main, because it is only applicable in keras
    
    * remove tailing blank line
    
    * add test for synth data
    
    * add kwargs to init
    
    * add kwargs to function invokation
    
    * correctly pass kwargs
    
    * debug
    
    * debug
    
    * debug
    
    * fix super init
    
    * bug fix
    
    * fix local_flags
    
    * fix import
    
    * bug fix
    
    * fix log_steps flag
    
    * bug fix
    
    * bug fix: add missing return value
    
    * resolve double-defined flags
    
    * lint fix
    
    * move log_steps flag to benchmarK flag
    
    * fix lint
    
    * lint fix
    
    * lint fix
    
    * try flag core default values
    
    * bug fix
    
    * bug fix
    
    * bug fix
    
    * debug
    
    * debug
    
    * remove debug prints
    
    * rename benchmark methods
    
    * flag bug fix for synth benchmark
    4c11b84b
keras_utils.py 2.53 KB