Python run_mcts示例

示例#1

try:
    parser = argparse.ArgumentParser(
        description='Run simple model against guac')
    parser.add_argument('objective',
                        type=int,
                        help="Guacamol objective index to target")
    parser.add_argument('--attempt',
                        help="Attempt number (used for multiple runs)",
                        default='')

    args = parser.parse_args()
    obj_num = args.objective
    attempt = args.attempt
except:
    obj_num = 0
    attempt = ''

ver = 'v2'
run_mcts(
    kind='model_thompson',
    compress_data_store=True,
    root_name='MCTSThompsonModel' + ver + '_obj' + str(obj_num) + '_attempt' +
    attempt,
    obj_num=obj_num,
    ver=ver,
    reset_cache=False,
    penalize_repetition=False,
    batch_size=70,
    num_batches=30,
)

示例#2

文件： run_mixed_model_mcts.py 项目： iisuslik43/generative_playground

import os, inspect
my_location = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
sys.path.append('../../../..')
from generative_playground.models.problem.mcts.mcts import run_mcts
from generative_playground.models.problem.mcts.node import MCTSNodeLocalThompson, MCTSNodeGlobalThompson
import argparse

try:
    parser = argparse.ArgumentParser(description='Run simple model against guac')
    parser.add_argument('objective', type=int, help="Guacamol objective index to target")
    parser.add_argument('--attempt', help="Attempt number (used for multiple runs)", default='')

    args = parser.parse_args()
    obj_num = args.objective
    attempt = args.attempt
except:
    obj_num = 0
    attempt = ''

ver = 'v2'
run_mcts(kind='model_mixed',
         compress_data_store=True,
         root_name='MCTSMixedModel' + ver + '_obj' + str(obj_num) + '_attempt' + attempt,
         obj_num=obj_num,
         ver=ver,
         reset_cache=False,
         penalize_repetition=True,
         decay=0.95,
         updates_to_refresh=10,
         batch_size=70,
         num_batches=30)

示例#3

文件： run_local_mcts.py 项目： iisuslik43/generative_playground

import os, sys, inspect
my_location = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
sys.path.append('../../../..')
from generative_playground.models.problem.mcts.mcts import run_mcts
from generative_playground.models.problem.mcts.node import MCTSNodeLocalThompson, MCTSNodeGlobalThompson

run_mcts(num_batches=10000,
         kind='thompson_local',
         compress_data_store=True, # not compressing is 50% faster but really disk-hungry
         base_name='MCTS_local',
         ver='v2',
         obj_num=0,
         reset_cache=True)

示例#4

try:
    parser = argparse.ArgumentParser(
        description='Run simple model against guac')
    parser.add_argument('objective',
                        type=int,
                        help="Guacamol objective index to target")
    parser.add_argument('--attempt',
                        help="Attempt number (used for multiple runs)",
                        default='')

    args = parser.parse_args()
    obj_num = args.objective
    attempt = args.attempt
except:
    obj_num = 0
    attempt = ''

ver = 'v2'
run_mcts(kind='model_global',
         compress_data_store=True,
         root_name='MCTSGlobalModel' + ver + '_obj' + str(obj_num) +
         '_attempt' + attempt,
         obj_num=obj_num,
         ver=ver,
         reset_cache=False,
         penalize_repetition=True,
         batch_size=5,
         num_batches=30,
         entropy_weight=0)

示例#5

文件： run_global_mcts.py 项目： iisuslik43/generative_playground

from generative_playground.models.problem.mcts.mcts import run_mcts
from generative_playground.models.problem.mcts.node import MCTSNodeLocalThompson, MCTSNodeGlobalThompson

run_mcts(num_batches=10000,
         node_type=MCTSNodeGlobalThompson,
         compress_data_store=False,
         dashboard_name='MCTS_global')