PAZ clean up: sort out license + publication dir + README

ukuhl · May 6, 2022 · dc9de12 · dc9de12
1 parent 93c3c04
commit dc9de12
Show file tree

Hide file tree

Showing 4 changed files with 419 additions and 26 deletions.
diff --git a/BackEnd/models.py b/BackEnd/models.py
@@ -31,7 +31,7 @@ def build_model(file_path="modelData/AlienZooDataSet_PAZ.csv"):
     print(y.shape)
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
     # Save dataset
-    np.savez("dataset.npz", X_train=X_train, X_test=X_test, y_train=y_train, y_test=y_test)
+    # np.savez("dataset.npz", X_train=X_train, X_test=X_test, y_train=y_train, y_test=y_test)
 
     # Fit model
     model = DecisionTreeRegressor(max_depth=4, random_state=42)