diff --git a/models/ai_models/nav_model.py b/models/ai_models/nav_model.py new file mode 100644 index 0000000..81f093c --- /dev/null +++ b/models/ai_models/nav_model.py @@ -0,0 +1,13 @@ +import pandas as pd +from sklearn.ensemble import RandomForestClassifier + +class NavModel: + def __init__(self, training_data): + self.training_data = training_data + self.model = RandomForestClassifier() + + def train(self): + self.model.fit(self.training_data.drop("target", axis=1), self.training_data["target"]) + + def predict(self, input_data): + return self.model.predict(input_data)