Skip to content

Instantly share code, notes, and snippets.

@sararob

sararob/format-labels.py Secret

Created August 14, 2018 18:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save sararob/2b1a63196200b4479d33ae80a9abd4c0 to your computer and use it in GitHub Desktop.
Save sararob/2b1a63196200b4479d33ae80a9abd4c0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
format-labels.py
# Genre lookup, each genre corresponds to an index
top_genres = ['Comedy', 'Thriller', 'Romance', 'Action', 'Horror', 'Crime', 'Documentary', 'Adventure', 'Science Fiction']
# Multi-hot label for an action and adventure movie
[0 0 0 1 0 0 0 1 0]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment