From 45df11a0e46e77c0ef6696b50457ba79e66acf79 Mon Sep 17 00:00:00 2001
From: Abhinav Behal <abhinav.behal@hotmail.com>
Date: Mon, 27 May 2019 23:51:43 +1200
Subject: [PATCH] Updated lambda with early stopping rounds parameter

---
 lambda/run_xgboost.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lambda/run_xgboost.py b/lambda/run_xgboost.py
index fc0e2f1..14d716d 100644
--- a/lambda/run_xgboost.py
+++ b/lambda/run_xgboost.py
@@ -23,6 +23,7 @@ def run(event, context):
     nfold = input_body['cv']
     seed = input_body['seed']
     num_boost_round = input_body['num_boost_round']
+    early_stopping_rounds = input_body['early_stopping_rounds']
 
     logger.info('Reading CSV')
     df = pd.read_csv('./preprocessed.csv')
@@ -40,6 +41,7 @@ def run(event, context):
         nfold=nfold,
         seed=seed,
         num_boost_round=num_boost_round,
+        early_stopping_rounds=early_stopping_rounds,
         metrics='error')
 
     logger.info("Done")