feat: add gender base models architectures
This commit is contained in:
@@ -8,3 +8,4 @@ models/
|
|||||||
.env.local
|
.env.local
|
||||||
var/
|
var/
|
||||||
/dataset/
|
/dataset/
|
||||||
|
.DS_Store
|
||||||
|
|||||||
@@ -8,6 +8,10 @@ from typing import Optional
|
|||||||
ROOT_DIR = os.getcwd()
|
ROOT_DIR = os.getcwd()
|
||||||
DATA_DIR = os.path.join(ROOT_DIR, 'dataset')
|
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
|
||||||
TRAINING_EPOCHS = 5
|
TRAINING_EPOCHS = 5
|
||||||
MODEL_NAME = f"./models/ners-{datetime.now().strftime('%Y%m%d%H%M%S')}"
|
MODEL_NAME = f"./models/ners-{datetime.now().strftime('%Y%m%d%H%M%S')}"
|
||||||
|
|||||||
@@ -0,0 +1,44 @@
|
|||||||
|
absl-py==2.3.0
|
||||||
|
astunparse==1.6.3
|
||||||
|
certifi==2025.6.15
|
||||||
|
charset-normalizer==3.4.2
|
||||||
|
flatbuffers==25.2.10
|
||||||
|
gast==0.6.0
|
||||||
|
google-pasta==0.2.0
|
||||||
|
grpcio==1.73.0
|
||||||
|
h5py==3.14.0
|
||||||
|
idna==3.10
|
||||||
|
joblib==1.5.1
|
||||||
|
keras==3.10.0
|
||||||
|
libclang==18.1.1
|
||||||
|
Markdown==3.8.2
|
||||||
|
markdown-it-py==3.0.0
|
||||||
|
MarkupSafe==3.0.2
|
||||||
|
mdurl==0.1.2
|
||||||
|
ml-dtypes==0.3.2
|
||||||
|
namex==0.1.0
|
||||||
|
numpy==1.26.4
|
||||||
|
opt_einsum==3.4.0
|
||||||
|
optree==0.16.0
|
||||||
|
packaging==25.0
|
||||||
|
pandas==2.3.0
|
||||||
|
protobuf==4.25.8
|
||||||
|
Pygments==2.19.1
|
||||||
|
python-dateutil==2.9.0.post0
|
||||||
|
pytz==2025.2
|
||||||
|
requests==2.32.4
|
||||||
|
rich==14.0.0
|
||||||
|
scikit-learn==1.7.0
|
||||||
|
scipy==1.15.3
|
||||||
|
six==1.17.0
|
||||||
|
tensorboard==2.16.2
|
||||||
|
tensorboard-data-server==0.7.2
|
||||||
|
tensorflow==2.16.2
|
||||||
|
tensorflow-io-gcs-filesystem==0.37.1
|
||||||
|
termcolor==3.1.0
|
||||||
|
threadpoolctl==3.6.0
|
||||||
|
typing_extensions==4.14.0
|
||||||
|
tzdata==2025.2
|
||||||
|
urllib3==2.5.0
|
||||||
|
Werkzeug==3.1.3
|
||||||
|
wrapt==1.17.2
|
||||||
Reference in New Issue
Block a user