【强化学习】pysc2 星际争霸环境测试

最新推荐文章于 2025-10-29 05:11:25 发布

原创

最新推荐文章于 2025-10-29 05:11:25 发布 · 414 阅读

0 ·

CC 4.0 BY-SA版权

这段代码定义了一个`Game_StarCraftII`类，用于创建和运行星际争霸II的环境。它接受配置参数，如玩家类型（智能体或电脑）、种族、难度等，并设置环境。环境初始化后，它会根据配置创建玩家，然后运行游戏循环，记录观察、行动和奖励。最后，它打印出游戏历史中的奖励。在`test.py`中，配置了一个`Config`对象并创建了包含一个`BaseAgent`的代理来运行游戏。

如何安装就不说了

Game.py

from pysc2.env import sc2_env

class Game_StarCraftII(object):
    def __init__(self, config, agents):
        self.game_history = {
   
   
            'observations': [],
            'actions': [],
            'rewards': None
        }
        players = []
        if config.player_one_type == 'Agent':
            player_one = sc2_env.Agent(
                race=sc2_env.Race[config.player_one_race]
            )
        elif config.player_one_type == 'Bot':
            player_one = sc2_env.Bot(
                race=sc2_env.Race[config.player_one_race],
                difficulty=sc2_env.Difficulty[config.difficulty]
            )
        else:
            raise NotImplementedError(