Update predictor.py

predictor.py

@@ -104,10 +104,32 @@ model = lgb.train(params,
                  callbacks = [early_stopping(300)]
                  )
 
+#Unoptimized (hand_picked)
+params = {
+    'metric': 'rmse',
+    'boosting_type' : 'gbdt',
+    'objective' : 'regression',
+    'max_depth' : 2,
+    'n_estimators' : 10000,
+    'num_leaves' : 4,
+    'verbose' : 1,
+    'learning_rate': 0.01,
+    'min_data_in_leaf': 100
+}
+
+
+lgbm = lgb.train(params,
+                 training_data,
+                 valid_sets=valid_data,
+                 early_stopping_rounds=3000
+                 )
+
 #function to call for predictions
 def make_a_prediction(X):
-    prediction = model.predict(X)
-    return prediction
+    opt_prediction = model.predict(X)
+    unopt_prediction = lgbm.predict(X)
+    predictions = [opt_prediction, unopt_prediction]
+    return predictions
 
 predictions = make_a_prediction(X_test)
 print(predictions)