Kaggle ConnectX Models
这个仓库包含了用于 Kaggle ConnectX 竞赛的强化学习模型
模型文件
alpha-zero-high.pth: AlphaZero 高性能模型alpha-zero-medium.pth: AlphaZero 中等性能模型DQN-base.pth: Deep Q-Network 基础模型
使用方法
import torch
# 加载模型
model = torch.load('alpha-zero-high.pth')
model.eval()
模型说明
AlphaZero Models
基于 AlphaZero 算法的模型,使用蒙特卡洛树搜索(MCTS)和深度神经网络。
DQN Model
基于 Deep Q-Network 的强化学习模型。