Views
No views yet
pipeline with the task sentiment-analysis. It should always give random results (i.e. {"label": "negative", "score": 0.5}).1classifier = pipeline("sentiment-analysis", "dhpollack/distilbert-dummy-sentiment")
2results = classifier(["this is a test", "another test"])[UNK]
[SEP]
[PAD]
[CLS]
[MASK]1import transformers
2config = transformers.DistilBertConfig(vocab_size=5, n_layers=1, n_heads=1, dim=1, hidden_dim=4 * 1, num_labels=2, id2label={0: "negative", 1: "positive"}, label2id={"negative": 0, "positive": 1})
3model = transformers.DistilBertForSequenceClassification(config)
4tokenizer = transformers.DistilBertTokenizer("/tmp/vocab.txt", model_max_length=512)
5config.save_pretrained(".")
6model.save_pretrained(".")
7tokenizer.save_pretrained(".")