Commit 19f8bd8e authored by 赵威's avatar 赵威

update learning rate for diary

parent f9425c62
......@@ -50,7 +50,7 @@ def main():
train_df, val_df = train_test_split(train_df, test_size=0.2)
all_features = fe.build_features(df, diary_fe.INT_COLUMNS, diary_fe.FLOAT_COLUMNS, diary_fe.CATEGORICAL_COLUMNS)
params = {"feature_columns": all_features, "hidden_units": [360, 200, 80, 2], "learning_rate": 0.1}
params = {"feature_columns": all_features, "hidden_units": [360, 200, 80, 2], "learning_rate": 0.2}
model_path = str(Path("/data/files/model_tmp/diary/").expanduser())
if os.path.exists(model_path):
shutil.rmtree(model_path)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment