Hugging Face's logo

Stanford-TH
/
GenrePrediction

Feature Extraction
Transformers
Safetensors
custom-bert-base-uncased
custom_code
Model card Files
xet
Community
GenrePrediction
File size: 1,120 Bytes
d5249c9
a6ed790
d5249c9
 
 
 
a6ed790
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d5249c9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a6ed790
d5249c9
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
{
  "_name_or_path": "Stanford-TH/GenrePrediction",
  "architectures": [
    "GenreModel"
  ],
  "auto_map": {
    "AutoConfig": "Stanford-TH/GenrePrediction--genre_configuration.GenreConfig",
    "AutoModel": "Stanford-TH/GenrePrediction--genre_model.GenreModel"
  },
  "custom_pipelines": {
    "weighted-genre-classification": {
      "default": {
        "model": {
          "pt": "Stanford-TH/GenrePrediction"
        }
      },
      "impl": "genre_pipe.GenrePredictionPipeline",
      "pt": [
        "AutoModel"
      ],
      "tf": [],
      "type": "text"
    }
  },
  "id2label": {
    "0": "Action",
    "1": "Adventure",
    "10": "Horror",
    "11": "Musical",
    "12": "Mystery",
    "13": "Romance",
    "14": "Sci-Fi",
    "15": "Sport",
    "16": "Thriller",
    "2": "Biography",
    "3": "Comedy",
    "4": "Crime",
    "5": "Documentary",
    "6": "Drama",
    "7": "Family",
    "8": "Fantasy",
    "9": "History"
  },
  "label2id": null,
  "model_type": "custom-bert-base-uncased",
  "torch_dtype": "float32",
  "transformers_version": "4.39.3"
}