-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
config #2
Comments
base_config: ../config/ dataset_config: tuner_space: |
base_config: ../config/ dataset_config: tuner_space: |
base_config: ../config/
base_expid: MLPX_base
dataset_id: avazu_x1
dataset_config:
avazu_x1:
data_root: ../data/Avazu/
data_format: csv
train_data: ../data/Avazu/Avazu_x1/train.csv
valid_data: ../data/Avazu/Avazu_x1/valid.csv
test_data: ../data/Avazu/Avazu_x1/test.csv
min_categr_count: 1
feature_cols:
- {name: ["feat_1", "feat_2", "feat_3", "feat_4", "feat_5", "feat_6", "feat_7", "feat_8", "feat_9", "feat_10",
"feat_11", "feat_12", "feat_13", "feat_14", "feat_15", "feat_16", "feat_17", "feat_18", "feat_19",
"feat_20", "feat_21", "feat_22"],
active: True, dtype: float, type: categorical}
label_col: {name: label, dtype: float}
tuner_space:
model_root: './Avazu/MLPX_avazu_x1/'
embedding_dim: 10
embedding_regularizer: 5.e-2
dnn1_batch_norm: True
dnn2_batch_norm: True
dnn1_hidden_units: [[400, 400, 400]]
dnn2_hidden_units: [[800]]
dnn1_dropout: [0.4, 0.3]
dnn2_dropout: [0.2, 0.3]
use_feature_selection: True
selection_hidden_units: [[1000]]
bilinear_group: [1, 2, 5, 10, 20, 50]
learning_rate: 1.e-3
batch_size: 4096
seed: 2021
monitor: 'AUC'
monitor_mode: 'max'
metrics: [['AUC', 'logloss']]
verbose: 0
The text was updated successfully, but these errors were encountered: