/spotify-tensorflow

Provides Spotify specific TensorFlow helpers

Primary LanguagePythonApache License 2.0Apache-2.0

spotify-tensorflow

Build Status codecov.io GitHub license PyPI version

Raison d'être:

Provide Spotify specific TensorFlow helpers.

Features

  • 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

Examples:

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 and test-requirements.txt are installed (probably in a virtualenv).
  • Invoke via scripts/run_examples.sh