【强化学习】强化学习环境搭建，gym构造五子棋Gomoku

最新推荐文章于 2025-09-12 13:59:50 发布

原创

最新推荐文章于 2025-09-12 13:59:50 发布 · 1.3k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习 #人工智能 #深度学习 #强化学习

这个代码实现了一个基于gym库的Go棋盘游戏环境。游戏环境包括初始化、重置、渲染和步进操作，并检查游戏是否结束。玩家可以在3xN的棋盘上下棋，目标是形成连续的五个棋子。环境使用二维数组表示棋盘状态，并通过渲染功能展示游戏界面。

from gym.envs.classic_control import rendering
import numpy as np
import gym

def check(filter, state, size, filter_w, filter_h):
    done = False
    result = []
    for i in range(size - filter_h + 1):
        for j in range(size - filter_w + 1):
            input_block = state[i:i + filter_h, j:j + filter_w]
            result.append(np.sum(filter * input_block))

    for i in result:
        if i == 5:
            done =True
    return done

class Gomoku(gym.Env):
    def __init__(self, num_chess, block_size):

        if num_chess < 5:
            raise ValueError("The minimum checkerboard is 5.")

        self.board = None
        self.num_chess = num_chess
        self.winner = None

        self.block_size = block_size

        self.viewer = rendering.Viewer(
            self.num_chess * self.block_size,
            self.num_chess * self.block_size
        )

        self.player = None

    def reset(self):
        self.board = np.zeros([3, self.num_chess, self.num_chess])
        self.player = 0
        self.viewer.geoms