feat: add gender base models architectures

2025-06-20 16:38:48 +02:00
parent f454ba7938
commit 1d58e3ccc4
5 changed files with 49 additions and 0 deletions
@@ -8,6 +8,10 @@ from typing import Optional
 ROOT_DIR = os.getcwd()
 DATA_DIR = os.path.join(ROOT_DIR, 'dataset')

+MODELS_DIR = os.path.join(ROOT_DIR, 'models')
+GENDER_MODELS_DIR = os.path.join(MODELS_DIR, 'gender')
+NER_MODELS_DIR = os.path.join(MODELS_DIR, 'ner')
+
 # Training
 TRAINING_EPOCHS = 5
 MODEL_NAME = f"./models/ners-{datetime.now().strftime('%Y%m%d%H%M%S')}"