fix

2024-05-14 23:43:14 +02:00 · 2024-05-14 23:43:14 +02:00 · 5651fb92d1
commit 5651fb92d1
parent a597ba4d2a
1 changed files with 1 additions and 1 deletions
--- a/model.py
+++ b/model.py
@ -29,7 +29,7 @@ preprocessor = ColumnTransformer(
 pipeline = Pipeline(steps=[
    ('preprocessor', preprocessor),
    ('model', Sequential([
-        Dense(64, activation='relu', input_dim=4),
+        Dense(64, activation='relu', input_dim=5),
        Dense(64, activation='relu'),
        Dense(1)
    ]))