From bea12fea1acf2d219bf2fefec88653f4df93aae1 Mon Sep 17 00:00:00 2001
From: Dave Berenbaum <dave.berenbaum@gmail.com>
Date: Thu, 31 Aug 2023 09:19:17 +0000
Subject: [PATCH] Run experiments tuning random forest params

---
 dvc.lock    | 16 ++++++++--------
 params.yaml |  4 ++--
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/dvc.lock b/dvc.lock
index 245ca17b1..f527a98b6 100644
--- a/dvc.lock
+++ b/dvc.lock
@@ -57,14 +57,14 @@ stages:
       size: 1666
     params:
       params.yaml:
-        train.min_split: 0.01
-        train.n_est: 50
+        train.min_split: 8
+        train.n_est: 100
         train.seed: 20170428
     outs:
     - path: model.pkl
       hash: md5
-      md5: d1f6e055f7f5e2827fcfae68d9b64d4c
-      size: 1958115
+      md5: 63cd44fd46b7f76875dc9556a2000cb5
+      size: 10322743
   evaluate:
     cmd: python src/evaluate.py model.pkl data/features
     deps:
@@ -75,8 +75,8 @@ stages:
       nfiles: 2
     - path: model.pkl
       hash: md5
-      md5: d1f6e055f7f5e2827fcfae68d9b64d4c
-      size: 1958115
+      md5: 63cd44fd46b7f76875dc9556a2000cb5
+      size: 10322743
     - path: src/evaluate.py
       hash: md5
       md5: a1a59f55636170fb56e0c6afd3e28fa4
@@ -84,6 +84,6 @@ stages:
     outs:
     - path: eval
       hash: md5
-      md5: 80a081570c800c60b9b98ca4b3c91dd7.dir
-      size: 1292342
+      md5: ea452ca56ec7558c9387cfd5162514e7.dir
+      size: 1186897
       nfiles: 8
diff --git a/params.yaml b/params.yaml
index 954cc22b5..2c166f6a5 100644
--- a/params.yaml
+++ b/params.yaml
@@ -8,6 +8,6 @@ featurize:
 
 train:
   seed: 20170428
-  n_est: 50
-  min_split: 0.01
+  n_est: 100
+  min_split: 8