/
githubmirror
/
Agent4Rec
Обзор
Документация
Войти
/
githubmirror
/
Agent4Rec
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
main.py
31 строка
689 B
chenyuxin1999
init repo
12 окт 2023, 05:43
12 окт 2023, 05:43
9a08cc7
Код
Авторство
О чём код?
import numpy as np from tqdm import tqdm import os from parse import parse_args from simulation.utils import fix_seeds from simulation.avatar import Avatar from simulation.arena import Arena import wandb # load model if __name__ == '__main__': args = parse_args() # print(args) fix_seeds(args.seed) # set random seed if(args.use_wandb): wandb.init( # set the wandb project where this run will be logged project = "sandbox", name = args.simulation_name, group = args.dataset ) arena_ = Arena(args) arena_.excute() print('Simulation finished!') if(args.use_wandb): wandb.finish()