mirror of
https://github.com/MoonlitJolteon/frc-stat-predictor.git
synced 2025-11-02 22:05:03 +00:00
Forgot to commit in small portions again
This commit is contained in:
9
llm_integration/match_outcome_prediction.py
Normal file
9
llm_integration/match_outcome_prediction.py
Normal file
@ -0,0 +1,9 @@
|
||||
class MatchPredictor:
|
||||
"""Predicts match outcomes using LLM."""
|
||||
|
||||
def __init__(self, llm_model):
|
||||
self.llm_model = llm_model
|
||||
|
||||
def predict_outcome(self, blue_alliance_data, red_alliance_data):
|
||||
prompt = f"Given blue alliance data: {blue_alliance_data} and red alliance data: {red_alliance_data}, predict the match outcome."
|
||||
return self.query_ollama(prompt)
|
||||
Reference in New Issue
Block a user