Skip to content

Commit 4b7fb6c

Browse files
ParagEkbotenzw0301
andauthored
Apply suggestions from code review-2
Co-authored-by: Kento Nozawa <[email protected]>
1 parent 81f7e91 commit 4b7fb6c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

comet/comet_integration.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44
55
1. Initializes a Comet ML experiment for logging.
66
2. Loads the Breast Cancer dataset and splits it into training and testing sets.
7-
3. Defines an evaluation function using F1-score, precision, and recall.
7+
3. Defines an evaluation function using accuracy.
88
4. Implements an Optuna objective function to optimize hyperparameters
9-
(n_estimators and max_depth) for the Random Forest model.
9+
(min_samples_leaf, max_depth, and min_samples_split) for the Random Forest model.
1010
5. Uses Optuna to run multiple trials and identify the best hyperparameters.
1111
6. Trains the final Random Forest model using the best-found hyperparameters.
1212
7. Logs training and testing metrics to Comet ML.

0 commit comments

Comments
 (0)