1. 20 Apr, 2019 1 commit
    • Shining Sun's avatar
      Remove contrib imports, or move them inline (#6591) · 8ff9eb54
      Shining Sun authored
      * Remove contrib imports, or move them inline
      
      * Use exposed API for FixedLenFeature
      
      * Replace tf.logging with absl logging
      
      * Change GFile to v2 APIs
      
      * replace tf.logging with absl loggin in movielens
      
      * Fixing an import bug
      
      * Change gfile to v2 APIs in code
      
      * Swap to keras optimizer v2
      
      * Bug fix for optimizer
      
      * Change tf.log to tf.keras.backend.log
      
      * Change the loss function to keras loss
      
      * convert another loss to keras loss
      
      * Resolve comments and fix lint
      
      * Add a doc string
      
      * Fix existing tests and add new tests for DS
      
      * Added tests for multi-replica
      
      * Fix lint
      
      * resolve comments
      
      * make estimator run in tf2.0
      
      * use compat v1 loss
      
      * fix lint issue
      8ff9eb54
  2. 01 Nov, 2018 1 commit
  3. 17 Sep, 2018 1 commit
  4. 30 Jul, 2018 1 commit
    • Taylor Robie's avatar
      NCF pipeline refactor (take 2) and initial TPU port. (#4935) · 6518c1c7
      Taylor Robie authored
      * intermediate commit
      
      * ncf now working
      
      * reorder pipeline
      
      * allow batched decode for file backed dataset
      
      * fix bug
      
      * more tweaks
      
      * parallize false negative generation
      
      * shared pool hack
      
      * workers ignore sigint
      
      * intermediate commit
      
      * simplify buffer backed dataset creation to fixed length record approach only. (more cleanup needed)
      
      * more tweaks
      
      * simplify pipeline
      
      * fix misplaced cleanup() calls. (validation works\!)
      
      * more tweaks
      
      * sixify memoryview usage
      
      * more sixification
      
      * fix bug
      
      * add future imports
      
      * break up training input pipeline
      
      * more pipeline tuning
      
      * first pass at moving negative generation to async
      
      * refactor async pipeline to use files instead of ipc
      
      * refactor async pipeline
      
      * move expansion and concatenation from reduce worker to generation workers
      
      * abandon complete async due to interactions with the tensorflow threadpool
      
      * cleanup
      
      * remove per...
      6518c1c7
  5. 25 Jun, 2018 1 commit
  6. 20 Jun, 2018 1 commit
    • Taylor Robie's avatar
      Wide Deep refactor and deep movies (#4506) · 20070ca4
      Taylor Robie authored
      * begin branch
      
      * finish download script
      
      * rename download to dataset
      
      * intermediate commit
      
      * intermediate commit
      
      * misc tweaks
      
      * intermediate commit
      
      * intermediate commit
      
      * intermediate commit
      
      * delint and update census test.
      
      * add movie tests
      
      * delint
      
      * fix py2 issue
      
      * address PR comments
      
      * intermediate commit
      
      * intermediate commit
      
      * intermediate commit
      
      * finish wide deep transition to vanilla movielens
      
      * delint
      
      * intermediate commit
      
      * intermediate commit
      
      * intermediate commit
      
      * intermediate commit
      
      * fix import
      
      * add default ncf csv construction
      
      * change default on download_if_missing
      
      * shard and vectorize example serialization
      
      * fix import
      
      * update ncf data unittests
      
      * delint
      
      * delint
      
      * more delinting
      
      * fix wide-deep movielens serialization
      
      * address PR comments
      
      * add file_io tests
      
      * investigate wide-deep test failure
      
      * remove hard coded path and properly use flags.
      
      * address file_io test PR comments
      
      * missed a hash_bucked_size
      20070ca4