Update selfchess.py

selfchess.py

@@ -5,6 +5,7 @@ import chess
 import os
 import chess.engine as eng
 import torch.multiprocessing as mp
+import random
 
 # CONFIGURATION
 CONFIG = {
@@ -16,6 +17,7 @@ CONFIG = {
     "num_games": 3000,
     "stockfish_time_limit": 1.0,
     "search_depth": 1,
+    "epsilon": 0.2
 }
 
 device = CONFIG["device"]
@@ -147,7 +149,7 @@ def game_gen(engine_side):
             evaling = {}
             for move in board.legal_moves:
                 board.push(move)
-                evaling[move] = -search(board, depth=CONFIG["search_depth"], alpha=float('-inf'), beta=float('inf'))
+                evaling[move] = -search(board, depth=CONFIG["search_depth"], alpha=float('-inf'), beta=float('inf')) * random.uniform(1-CONFIG["epsilon"],1+CONFIG["epsilon"])
                 board.pop()
             
             if not evaling: