From bda2e1451e8b889df227a200900918a1121013a3 Mon Sep 17 00:00:00 2001
From: Chris Thurber <8137212+skiman6010@users.noreply.github.com>
Date: Mon, 16 Oct 2017 14:08:14 -0400
Subject: [PATCH] Update run.py

Removed n_iter in favor of the new max_iter. Performance and accuracy is consistent. The change was because n_iter is being deprecated and will be removed in sklearn 0.21 according the the deprecation warning.
---
 xeneta_qualifier/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/xeneta_qualifier/run.py b/xeneta_qualifier/run.py
index e98b8f3..419f878 100644
--- a/xeneta_qualifier/run.py
+++ b/xeneta_qualifier/run.py
@@ -26,7 +26,7 @@ def runForest(X_train, y_train):
 
 # Stochastic Gradient Descent Classifier
 def runSGD(X_train, y_train):
-    sgd = SGDClassifier(n_iter=500, loss='modified_huber', penalty='elasticnet', random_state=42)
+    sgd = SGDClassifier(max_iter=500, loss='modified_huber', penalty='elasticnet', random_state=42)
     sgd.fit(X_train, y_train)
     return sgd