def __init__(self): #Strategy.__init__(self,"Defenseur") dic_strategy = { FonceurStrategy().name: FonceurStrategy(), Milieu().name: Milieu(), Gardien().name: Gardien() } with open("tree_test.pkl", "rb") as f: dt = pickle.load(f) DTreeStrategy.__init__(self, dt, dic_strategy, my_get_features)
from footIA.strats import GoalStrategy, DribleStrategy, FonceurStrategy, MultipurposeStrategy, RandomStrategy from soccersimulator import SoccerTeam, Simulation, show_simu,KeyboardStrategy,DTreeStrategy,load_jsonz,dump_jsonz from soccersimulator import apprend_arbre, build_apprentissage, genere_dot from footIA import MultipurposeStrategy,GoalStrategy,ToolBox,DribleStrategy,DefenseurStrategy import sklearn import numpy as np import pickle from arbre import my_get_features dic_strategy = {MultipurposeStrategy().name:MultipurposeStrategy(),GoalStrategy().name:GoalStrategy(),DribleStrategy().name:DribleStrategy(),DefenseurStrategy().name:DefenseurStrategy()} with open("tree_test.pkl","rb") as f: dt = pickle.load(f) treeStrat1 = DTreeStrategy(dt,dic_strategy,my_get_features) ## Creation d'une equipe pyteam = SoccerTeam(name="PyTeam") control = SoccerTeam(name="controlStratTeam") pyteam.add("verstile", MultipurposeStrategy()) control.add("control", treeStrat1) #pyteam.add("dVersatile2", MultipurposeStrategy()) #Creation d'une partie simu = Simulation(pyteam,control) #Jouer et afficher la partie show_simu(simu)
#states_tuple1 = load_jsonz(fname1 + ".jz") ent.apprendre_1v1(states_tuple0, ent.my_get_features_1v1, fname0 + ".pkl") #ent.apprendre(states_tuple1,ent.my_get_features, fname1 + ".pkl") with open(fname0 + ".pkl", "rb") as f: dt0 = pickle.load(f) # Visualisation de l'arbre genere_dot(dt0, fname0 + "testeee.dot") #with open(fname1 + ".pkl","rb") as f: # dt1 = pickle.load(f) ##Visualisation de l'arbre #genere_dot(dt1,fname1 + "rtestee1.dot") treeStrat0 = DTreeStrategy(dt0, dic_strategy, ent.my_get_features_1v1) #treeStrat1 = DTreeStrategy(dt1, dic_strategy, ent.my_get_features) treeteam = SoccerTeam("Arbre Team") test = input('Wanna test the ia? ') if test == 'y': nb_ia = input('Against which team : ') nb_ia = int(nb_ia) team2 = list_ia[nb_ia].get_team(1) treeteam.add("Joueur 0", treeStrat0) # treeteam.add("Joueur 1",treeStrat1)
pickle.dump(dt, f) return dt if __name__ == "__main__": entrainer("test_kb_strat.jz") dic_strategy = {Fonceur().name: Fonceur(), Defenseur().name: Defenseur()} states_tuple = load_jsonz("test_kb_strat.jz") apprendre(states_tuple, fichier="tree_test.pkl") with open("tree_test.pkl", "rb") as f: dt = pickle.load(f) # Visualisation de l'arbre genere_dot(dt, "test_arbre.dot") #Utilisation de l'arbre : arbre de decision, dico strategy, fonction de transformation etat -> attributs treeStrat1 = DTreeStrategy(dt, dic_strategy, extraction_attributs) #Création des équipes treeteam = SoccerTeam(name="Arbre") team2 = SoccerTeam(name="Sparring") treeteam.add("Arbre 1", treeStrat1) treeteam.add("Arbre 2", treeStrat1) team2.add("Joueur 1", Fonceur()) team2.add("Joueur 2", Defenseur()) simu = Simulation(treeteam, team2) show_simu(simu)