pikaduck's picture
lfs added
6cf89af
raw
history blame contribute delete
546 Bytes
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')