File size: 546 Bytes
6cf89af
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
import os

PROJECT_DIR = os.path.dirname(os.path.abspath(__file__))

MODELS_DIR = os.path.join(PROJECT_DIR, 'models')

FLAIR_MODEL_PATH = os.path.join(MODELS_DIR, 'flair-sentiment-classifier/best-model.pt')
SPACY_MODEL_PATH = os.path.join(MODELS_DIR, 'spacy-classifier/model-best')
SETFIT_MODEL_PATH = os.path.join(MODELS_DIR, 'setfit-classifier')
NB_MODEL_PATH = os.path.join(MODELS_DIR, 'm_nb.sav')
COUNT_VECTORIZER_PATH = os.path.join(MODELS_DIR, 'count_vectorizer.sav')
TFIDF_VECTORIZER_PATH = os.path.join(MODELS_DIR, 'tfidf_vectorizer.sav')