Create random_forest.py

ai/models/random_forest.py

@@ -0,0 +1,18 @@
+import pandas as pd
+from sklearn.ensemble import RandomForestClassifier
+from sklearn.model_selection import train_test_split
+
+class RandomForestModel:
+    def __init__(self, n_estimators=100, max_depth=None):
+        self.model = RandomForestClassifier(n_estimators=n_estimators, max_depth=max_depth)
+
+    def train(self, X_train, y_train):
+        self.model.fit(X_train, y_train)
+
+    def predict(self, X_test):
+        return self.model.predict(X_test)
+
+    def evaluate(self, X_test, y_test):
+        y_pred = self.predict(X_test)
+        accuracy = accuracy_score(y_test, y_pred)
+        return accuracy