tmdb-genre-classifier / genre_meta.json
arjun-varma's picture
Upload genre_meta.json with huggingface_hub
4c90071 verified
{
"genre_names": [
"Action",
"Adventure",
"Animation",
"Comedy",
"Crime",
"Documentary",
"Drama",
"Family",
"Fantasy",
"History",
"Horror",
"Music",
"Mystery",
"Romance",
"Science Fiction",
"TV Movie",
"Thriller",
"War",
"Western"
],
"metrics": {
"precision_micro": 0.16293859649122808,
"recall_micro": 0.9828042328042328,
"f1_micro": 0.2795334838224229,
"precision_macro": 0.16054097478259735,
"recall_macro": 0.9834885327429802,
"f1_macro": 0.26924159809023623,
"n_train": 1067,
"n_test": 267,
"threshold": 0.25,
"per_genre_f1": {
"Action": 0.3873015873015873,
"Adventure": 0.30662020905923343,
"Animation": 0.20553359683794467,
"Comedy": 0.40894568690095845,
"Crime": 0.23703703703703705,
"Documentary": 0.18340611353711792,
"Drama": 0.5269121813031161,
"Family": 0.27007299270072993,
"Fantasy": 0.2074074074074074,
"History": 0.16988416988416988,
"Horror": 0.25,
"Music": 0.149812734082397,
"Mystery": 0.21561338289962825,
"Romance": 0.27956989247311825,
"Science Fiction": 0.24113475177304963,
"TV Movie": 0.26181818181818184,
"Thriller": 0.463855421686747,
"War": 0.13934426229508196,
"Western": 0.21132075471698114
}
}
}