Update app.py
Browse files
app.py
CHANGED
|
@@ -81,7 +81,7 @@ def format_insights(insights, visuals):
|
|
| 81 |
|
| 82 |
def analyze_data(csv_file, additional_notes=""):
|
| 83 |
|
| 84 |
-
|
| 85 |
learning_rate = trial.suggest_loguniform("learning_rate", 1e-5, 5e-3)
|
| 86 |
batch_size = trial.suggest_categorical("batch_size", [8, 16, 32])
|
| 87 |
num_epochs = trial.suggest_int("num_epochs", 1, 5)
|
|
|
|
| 81 |
|
| 82 |
def analyze_data(csv_file, additional_notes=""):
|
| 83 |
|
| 84 |
+
def objective(trial):
|
| 85 |
learning_rate = trial.suggest_loguniform("learning_rate", 1e-5, 5e-3)
|
| 86 |
batch_size = trial.suggest_categorical("batch_size", [8, 16, 32])
|
| 87 |
num_epochs = trial.suggest_int("num_epochs", 1, 5)
|