From 5b0853baa8ac656704ecb9bf133742771c2e2b5f Mon Sep 17 00:00:00 2001
From: Pinapelz <yukais@pinapelz.com>
Date: Sat, 16 Dec 2023 23:06:11 -0800
Subject: change regression model to ridge + limit projection dataset to past 3
 months

---
 backend/app.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/backend/app.py b/backend/app.py
index e042a35..b29ca17 100644
--- a/backend/app.py
+++ b/backend/app.py
@@ -7,7 +7,7 @@ from sql.sql_handler import SQLHandler
 import fileutil as fs
 import datetime
 import pandas
-from sklearn.linear_model import LinearRegression
+from sklearn.linear_model import Ridge
 import numpy as np
 
 app = Flask(__name__)
@@ -100,8 +100,11 @@ def get_channel_information(channel_name):
     df = pandas.DataFrame(data=data)
     df['dates'] = pandas.to_datetime(df['dates'])
     df.set_index('dates', inplace=True)
+    df.sort_index(inplace=True)
+    three_months_ago = datetime.datetime.now() - datetime.timedelta(days=90)
+    df = df[df.index > three_months_ago]
     try:
-        model = LinearRegression()
+        model = Ridge(alpha=100)
         X = np.array(range(len(df))).reshape(-1, 1)
         y = df['subscribers']
         model.fit(X, y)
-- 
cgit v1.2.3