pytorch训练五子棋ai

有3个文件

game.py  五子棋游戏

mod.py  神经网络模型

xl.py   训练的代码

aigame.py   玩家与对战的五子棋

game.py

 
class Game:
    def __init__(self, h, w):
        # 行数
        self.h = h
        # 列数
        self.w = w
        # 棋盘
        self.L = [['-' for _ in range(w)] for _ in range(h)]
        # 当前玩家 - 表示空 X先下 然后是O
        self.cur = 'X'
        # 游戏胜利者
        self.win_user = None
 
    # 检查下完这步后有没有赢 y是行 x是列 返回True表示赢
    def check_win(self, y, x):
        directions = [
            # 水平、垂直、两个对角线方向
            (1, 0), (0, 1), (1, 1), (1, -1)
        ]
        player = self.L[y][x]
        for dy, dx in directions:
            count = 0
            # 检查四个方向上的连续相同棋子
            for i in range(-4, 5):  # 检查-4到4的范围，因为五子连珠需要5个棋子
                ny, nx = y + i * dy, x + i * dx
                if 0 <= ny < self.h and 0 <= nx < self.w and self.L[ny][nx] == player:
                    count += 1
                    if count == 5:
                        return True
                else:
                    count = 0
        return False
 
    # 检查能不能下这里 y行 x列 返回True表示能下
    def check(self, y, x):
        return self.L[y][x] == '-' and self.win_user is None
 
    # 打印棋盘 可视化用得到
    def __str__(self):
        # 确定行号和列号的宽度
        row_width = len(str(self.h - 1))
        col_width = len(str(self.w - 1))
        
        # 生成带有行号和列号的棋盘字符串表示
        result = []
        # 添加列号标题
        result.append(' ' * (row_width + 1) + ' '.join(f'{i:>{col_width}}' for i in range(self.w)))
        # 添加分隔线（可选）
        result.append(' ' * (row_width + 1) + '-' * (col_width * self.w))
        # 添加棋盘行
        for y, row in enumerate(self.L):
            # 添加行号
            result.append(f'{y:>{row_width}} ' + ' '.join(f'{cell:>{col_width}}' for cell in row))
        return '\n'.join(re