| { | |
| "_name_or_path": "shhossain/all-MiniLM-L6-v2-sentiment-classifier", | |
| "architectures": [ | |
| "SententenceTransformerSentimentModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "shhossain/all-MiniLM-L6-v2-sentiment-classifier--model.SentimentConfig", | |
| "AutoModelForSequenceClassification": "shhossain/all-MiniLM-L6-v2-sentiment-classifier--model.SententenceTransformerSentimentModel" | |
| }, | |
| "class_map": { | |
| "0": "sad", | |
| "1": "joy", | |
| "2": "love", | |
| "3": "anger", | |
| "4": "fear", | |
| "5": "surprise" | |
| }, | |
| "custom_pipelines": { | |
| "text-classification": { | |
| "impl": "sentpipeline.SentimentModelPipe", | |
| "pt": [ | |
| "AutoModelForSequenceClassification" | |
| ], | |
| "tf": [], | |
| "type": "text" | |
| } | |
| }, | |
| "embedding_model": "sentence-transformers/all-MiniLM-L6-v2", | |
| "h1": 44, | |
| "h2": 46, | |
| "model_type": "SententenceTransformerSentimentClassifier", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.29.0" | |
| } | |