bernard-ng/drc-ners-nlp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: use full_name feature for all models

Browse Source
This commit is contained in:
bernard-ng
2025-08-19 19:36:04 +02:00
parent 7101cea5e7
commit 0816207a2c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/research_templates.yaml
+4 -4
View File
@@ -26,7 +26,7 @@ baseline_experiments:
- name: "ensemble" - name: "ensemble"
description: "Baseline Ensemble with multiple models" description: "Baseline Ensemble with multiple models"
model_type: "ensemble" model_type: "ensemble"
features: [ "full_name", "name_length", "word_count" ] features: [ "full_name" ]
model_params: model_params:
base_models: [ "logistic_regression", "random_forest", "xgboost" ] base_models: [ "logistic_regression", "random_forest", "xgboost" ]
voting: "soft" voting: "soft"
@@ -36,7 +36,7 @@ baseline_experiments:
- name: "lightgbm" - name: "lightgbm"
description: "Baseline LightGBM with engineered features" description: "Baseline LightGBM with engineered features"
model_type: "lightgbm" model_type: "lightgbm"
features: [ "full_name", "name_length", "word_count" ] features: [ "full_name" ]
model_params: model_params:
n_estimators: 100 n_estimators: 100
max_depth: -1 max_depth: -1
@@ -92,7 +92,7 @@ baseline_experiments:
- name: "random_forest" - name: "random_forest"
description: "Baseline Random Forest with engineered features" description: "Baseline Random Forest with engineered features"
model_type: "random_forest" model_type: "random_forest"
features: [ "name_length", "word_count", "province" ] features: [ "full_name" ]
model_params: model_params:
n_estimators: 100 n_estimators: 100
max_depth: 10 max_depth: 10
@@ -126,7 +126,7 @@ baseline_experiments:
- name: "xgboost" - name: "xgboost"
description: "Baseline XGBoost with engineered features" description: "Baseline XGBoost with engineered features"
model_type: "xgboost" model_type: "xgboost"
features: [ "full_name", "name_length", "word_count" ] features: [ "full_name" ]
model_params: model_params:
n_estimators: 100 n_estimators: 100
max_depth: 6 max_depth: 6