fix: notebooks
This commit is contained in:
@@ -73,37 +73,6 @@ baseline_experiments:
|
||||
batch_size: 32
|
||||
tags: [ "baseline", "neural", "cnn", "surname" ]
|
||||
|
||||
## Ensemble Models
|
||||
- name: "ensemble"
|
||||
description: "Baseline Ensemble with multiple models"
|
||||
model_type: "ensemble"
|
||||
features: [ "full_name" ]
|
||||
model_params:
|
||||
base_models: [ "logistic_regression", "random_forest", "xgboost" ]
|
||||
voting: "soft"
|
||||
cv_folds: 5
|
||||
tags: [ "baseline", "ensemble" ]
|
||||
|
||||
- name: "ensemble_native"
|
||||
description: "Baseline Ensemble with native name"
|
||||
model_type: "ensemble"
|
||||
features: [ "native_name" ]
|
||||
model_params:
|
||||
base_models: [ "logistic_regression", "random_forest", "xgboost" ]
|
||||
voting: "soft"
|
||||
cv_folds: 5
|
||||
tags: [ "baseline", "ensemble", "native" ]
|
||||
|
||||
- name: "ensemble_surname"
|
||||
description: "Baseline Ensemble with surname"
|
||||
model_type: "ensemble"
|
||||
features: [ "surname" ]
|
||||
model_params:
|
||||
base_models: [ "logistic_regression", "random_forest", "xgboost" ]
|
||||
voting: "soft"
|
||||
cv_folds: 5
|
||||
tags: [ "baseline", "ensemble", "surname" ]
|
||||
|
||||
# LightGBM Models
|
||||
- name: "lightgbm"
|
||||
description: "Baseline LightGBM with engineered features"
|
||||
@@ -262,40 +231,6 @@ baseline_experiments:
|
||||
min_samples_leaf: 1
|
||||
tags: [ "baseline", "random_forest", "engineered", "surname" ]
|
||||
|
||||
# SVM Models
|
||||
- name: "svm"
|
||||
description: "Baseline SVM with full name features"
|
||||
model_type: "svm"
|
||||
features: [ "full_name" ]
|
||||
model_params:
|
||||
C: 1.0
|
||||
kernel: "rbf"
|
||||
ngram_range: [ 2, 4 ]
|
||||
max_features: 5000
|
||||
tags: [ "baseline", "svm" ]
|
||||
|
||||
- name: "svm_native"
|
||||
description: "Baseline SVM with native name features"
|
||||
model_type: "svm"
|
||||
features: [ "native_name" ]
|
||||
model_params:
|
||||
C: 1.0
|
||||
kernel: "rbf"
|
||||
ngram_range: [ 2, 4 ]
|
||||
max_features: 5000
|
||||
tags: [ "baseline", "svm", "native" ]
|
||||
|
||||
- name: "svm_surname"
|
||||
description: "Baseline SVM with surname features"
|
||||
model_type: "svm"
|
||||
features: [ "surname" ]
|
||||
model_params:
|
||||
C: 1.0
|
||||
kernel: "rbf"
|
||||
ngram_range: [ 2, 4 ]
|
||||
max_features: 5000
|
||||
tags: [ "baseline", "svm", "surname" ]
|
||||
|
||||
# Transformer Models
|
||||
- name: "transformer"
|
||||
description: "Baseline Transformer with attention mechanism"
|
||||
|
||||
Reference in New Issue
Block a user