Update evaluation classes

This commit is contained in:
2025-08-25 20:22:31 -04:00
parent cbe16aba8a
commit d7d6263cc1

View File

@@ -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]
return {
"depth": self.depth,
"window": self.window,
}