three layers for tractate

src/train_tractate.py

@@ -47,7 +47,7 @@ def main():
     train_df, val_df = train_test_split(train_df, test_size=0.2)
 
     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": [128, 64, 32], "learning_rate": 0.1}
     model_path = str(Path("/data/files/model_tmp/tractate/").expanduser())
     if os.path.exists(model_path):
         shutil.rmtree(model_path)