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)