Provide Spotify specific TensorFlow helpers.
- tf.data integration with Featran
- common Dataset API to read:
- TFRecord datasets as tf.Tensor
- TFRecord datasets as Pandas DataFrame
- TFRecord datasets as python dict
- tf.Graph freezing utilities
- TensorFlow integration with Luigi
Check examples in the examples directory, currently they include:
- sklearn training via Pandas DataFrame dataset
- XGBoost training via tf.Example dataset
- TensorFlow training via tf.Example dataset
To run the examples:
- Ensure
requirements.txt
andtest-requirements.txt
are installed (probably in a virtualenv). - Invoke via
scripts/run_examples.sh