Instructions to use njprogrammer/goemotion-BERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use njprogrammer/goemotion-BERT with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("njprogrammer/goemotion-BERT") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - setfit
How to use njprogrammer/goemotion-BERT with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("njprogrammer/goemotion-BERT") preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"]) print(preds) - Notebooks
- Google Colab
- Kaggle
Download config.json from njprogrammer/goemotion-BERT: direct link, hf CLI and curl.
- Browser
- Download file 1.93 kB
-
https://huggingface.co/njprogrammer/goemotion-BERT/resolve/main/config.json
- Command line
-
hf download hf://njprogrammer/goemotion-BERT/config.json
-
curl -L -o config.json https://huggingface.co/njprogrammer/goemotion-BERT/resolve/main/config.json
1.93 kB
| { | |
| "_name_or_path": "C:\\Users\\njpark/.cache\\torch\\sentence_transformers\\monologg_bert-base-cased-goemotions-original", | |
| "architectures": [ | |
| "BertModel" | |
| ], | |
| "attention_probs_dropout_prob": 0.1, | |
| "classifier_dropout": null, | |
| "finetuning_task": "goemotions", | |
| "hidden_act": "gelu", | |
| "hidden_dropout_prob": 0.1, | |
| "hidden_size": 768, | |
| "id2label": { | |
| "0": "admiration", | |
| "1": "amusement", | |
| "2": "anger", | |
| "3": "annoyance", | |
| "4": "approval", | |
| "5": "caring", | |
| "6": "confusion", | |
| "7": "curiosity", | |
| "8": "desire", | |
| "9": "disappointment", | |
| "10": "disapproval", | |
| "11": "disgust", | |
| "12": "embarrassment", | |
| "13": "excitement", | |
| "14": "fear", | |
| "15": "gratitude", | |
| "16": "grief", | |
| "17": "joy", | |
| "18": "love", | |
| "19": "nervousness", | |
| "20": "optimism", | |
| "21": "pride", | |
| "22": "realization", | |
| "23": "relief", | |
| "24": "remorse", | |
| "25": "sadness", | |
| "26": "surprise", | |
| "27": "neutral" | |
| }, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 3072, | |
| "label2id": { | |
| "admiration": 0, | |
| "amusement": 1, | |
| "anger": 2, | |
| "annoyance": 3, | |
| "approval": 4, | |
| "caring": 5, | |
| "confusion": 6, | |
| "curiosity": 7, | |
| "desire": 8, | |
| "disappointment": 9, | |
| "disapproval": 10, | |
| "disgust": 11, | |
| "embarrassment": 12, | |
| "excitement": 13, | |
| "fear": 14, | |
| "gratitude": 15, | |
| "grief": 16, | |
| "joy": 17, | |
| "love": 18, | |
| "nervousness": 19, | |
| "neutral": 27, | |
| "optimism": 20, | |
| "pride": 21, | |
| "realization": 22, | |
| "relief": 23, | |
| "remorse": 24, | |
| "sadness": 25, | |
| "surprise": 26 | |
| }, | |
| "layer_norm_eps": 1e-12, | |
| "max_position_embeddings": 512, | |
| "model_type": "bert", | |
| "num_attention_heads": 12, | |
| "num_hidden_layers": 12, | |
| "pad_token_id": 0, | |
| "position_embedding_type": "absolute", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.30.2", | |
| "type_vocab_size": 2, | |
| "use_cache": true, | |
| "vocab_size": 28996 | |
| } | |