import os
import sys
while not os.getcwd().endswith(
"personality_prediction") and os.getcwd() != "/":
os.chdir(os.path.dirname(os.getcwd()))
if os.getcwd() == "/":
raise Exception(
"The project dir's name must be 'personality_prediction'. Rename it.")
sys.path.append(os.getcwd())
from utils import Config, load_yaml_config
config = Config()
config.ocean_traits = [0, 1, 2, 3, 4]
# OCEAN personality traits to which tune the embedding: O:0, C:1, E:2, A:3, N:4
config.epochs_number = 10
# NLP model's training epochs
config.num_reviews = 1500000
# number of reviews to use for training (training set + test set)
config.voc_dim = 6 * 10**4
# number of terms in the tuned embedding
config.train_zeros = False
# use True if you want to train weights representing padding's tokens, use False otherwise.
config.output_type = "mean"
# target of the model: 'mean' or 'sum' of known terms' scores in the review.
config.shuffle = True
# if True review from yelp dataset will be shuffled before extracting num_reviews reviews.
# if False the first num_reviews of yelp dataset will be extracted.
config.features_config = [100, int(100 / 2), int(100 / 4)]
# configuration of NLP model's architecture: features, filters and hidden units.
config.embedding_name = "new_tuned_embedding"
