diff --git a/scripts/evaluation.py b/scripts/evaluation.py index e95251e..2a3e6c8 100644 --- a/scripts/evaluation.py +++ b/scripts/evaluation.py @@ -1,10 +1,12 @@ import random -import ctypes as C from binding.python_c_ffi import ChessFFI from binding.python_c_ffi import Move class BaseEvaluation: + NAME = "" + + def __init__(self, chess_ffi=None): if chess_ffi is None: chess_ffi = ChessFFI() @@ -27,6 +29,9 @@ class BaseEvaluation: a simplistic approach. """ class RandomEval(BaseEvaluation): + NAME = "random" + + def __init__(self, chess_ffi=None): super().__init__(chess_ffi) @@ -36,6 +41,9 @@ class RandomEval(BaseEvaluation): class NegaMaxEval(BaseEvaluation): + NAME = "negamax" + + def __init__(self, depth=6, cp_window=20, chess_ffi=None): super().__init__(chess_ffi) self.depth = depth @@ -57,4 +65,7 @@ class NegaMaxEval(BaseEvaluation): def get_params(self): - return [self.depth, self.window] \ No newline at end of file + return { + "depth": self.depth, + "window": self.window, + } \ No newline at end of file