Create ai_service.py

KOSASIH · Aug 7, 2024 · 1e78cdc · 1e78cdc
1 parent 939895a
commit 1e78cdc
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/blockchain_integration/pi_network/pinnacle/src/services/ai_service.py b/blockchain_integration/pi_network/pinnacle/src/services/ai_service.py
@@ -0,0 +1,14 @@
+from models.ai_model import AIModel
+
+class AIService:
+    def __init__(self, ai_model: AIModel):
+        self.ai_model = ai_model
+
+    def predict(self, data: pd.DataFrame) -> pd.Series:
+        return self.ai_model.predict(data)
+
+    def evaluate(self, data: pd.DataFrame) -> float:
+        return self.ai_model.evaluate(data)
+
+    def train(self, data: pd.DataFrame) -> None:
+        self.ai_model.train(data)