Update model hyperparameters and number of epochs

app.py

@@ -32,7 +32,7 @@ class DNN(nn.Module):
 
 
 # Load the model
-model = DNN(input_size=7, hidden_size=256, output_size=1)
+model = DNN(input_size=7, hidden_size=128, output_size=1)
 
 # Initialize the OneHotEncoder
 encoder = OneHotEncoder(handle_unknown="ignore")
@@ -94,10 +94,10 @@ async def train(file: UploadFile = File(...)):
     optimizer = optim.Adam(model.parameters(), lr=0.0001)
 
     # Fit the model
-    kf = KFold(n_splits=5)
+    kf = KFold(n_splits=4)
     accuracies = []
 
-    epochs = 50  # Define the number of epochs
+    epochs = 25  # Define the number of epochs
 
     for epoch in range(epochs):
         for train_index, test_index in kf.split(X):