shhossain commited on
Commit
9706db3
1 Parent(s): 8e48215

Delete config.py

Browse files
Files changed (1) hide show
  1. config.py +0 -15
config.py DELETED
@@ -1,15 +0,0 @@
1
- from transformers import PretrainedConfig
2
-
3
-
4
- mp = {0:'sad',1:'joy',2:'love',3:'anger',4:'fear',5:'surprise'}
5
-
6
- class SentimentConfig(PretrainedConfig):
7
- model_type = "SententenceTransformerSentimentClassifier"
8
-
9
- def __init__(self, embedding_model: str="sentence-transformers/all-MiniLM-L6-v2", class_map: dict=mp, h1: int=44, h2: int=46, **kwargs):
10
- self.embedding_model = embedding_model
11
- self.class_map = class_map
12
- self.h1 = h1
13
- self.h2 = h2
14
-
15
- super().__init__(**kwargs)