Commit a169c1b0 authored by 赵威's avatar 赵威

update tmp path

parent b29b88b9
...@@ -43,7 +43,7 @@ def main(): ...@@ -43,7 +43,7 @@ def main():
all_features = fe.build_features(df, diary_fe.INT_COLUMNS, diary_fe.FLOAT_COLUMNS, diary_fe.CATEGORICAL_COLUMNS) 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": [64, 32], "learning_rate": 0.1} params = {"feature_columns": all_features, "hidden_units": [64, 32], "learning_rate": 0.1}
model_path = str(Path("~/data/model_tmp/").expanduser()) model_path = str(Path("~/data/model_tmp/diary/").expanduser())
# if os.path.exists(model_path): # if os.path.exists(model_path):
# shutil.rmtree(model_path) # shutil.rmtree(model_path)
......
...@@ -40,7 +40,7 @@ def main(): ...@@ -40,7 +40,7 @@ def main():
all_features = fe.build_features(df, tractate_fe.INT_COLUMNS, tractate_fe.FLOAT_COLUMNS, tractate_fe.CATEGORICAL_COLUMNS) all_features = fe.build_features(df, tractate_fe.INT_COLUMNS, tractate_fe.FLOAT_COLUMNS, tractate_fe.CATEGORICAL_COLUMNS)
params = {"feature_columns": all_features, "hidden_units": [64, 32], "learning_rate": 0.1} params = {"feature_columns": all_features, "hidden_units": [64, 32], "learning_rate": 0.1}
model_path = str(Path("~/data/model_tmp/").expanduser()) model_path = str(Path("~/data/model_tmp/tractate/").expanduser())
# if os.path.exists(model_path): # if os.path.exists(model_path):
# shutil.rmtree(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