summaryrefslogtreecommitdiffstats
path: root/Src/PyCatcher/src/evaluators.py
diff options
context:
space:
mode:
Diffstat (limited to 'Src/PyCatcher/src/evaluators.py')
-rw-r--r--Src/PyCatcher/src/evaluators.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/Src/PyCatcher/src/evaluators.py b/Src/PyCatcher/src/evaluators.py
index 3170f4a..de624fa 100644
--- a/Src/PyCatcher/src/evaluators.py
+++ b/Src/PyCatcher/src/evaluators.py
@@ -4,6 +4,11 @@ class EvaluatorSelect:
CONSERVATIVE = 0
WEIGHTED = 1
BAYES = 2
+ MACHINE = 3
+
+class StationClass:
+ BASE_STATION = 0
+ CATCHER = 1
class Evaluator:
@@ -28,4 +33,7 @@ class BayesEvaluator(Evaluator):
return_type = type(int)
class WeightedEvaluator(Evaluator):
- return_type = type(int) \ No newline at end of file
+ return_type = type(int)
+
+class MachineLearningEvaluator(Evaluator):
+ return_type = type(StationClass) \ No newline at end of file