def apprendre(exemples, get_features, fname=None):
#genere l'ensemble d'apprentissage
data_train, data_labels = build_apprentissage(exemples, get_features)
## Apprentissage de l'arbre
dt = apprend_arbre(data_train,
data_labels,
depth=10,
feature_names=get_features.names)
##Sauvegarde de l'arbre
if fname is not None:
with open(fname, "wb") as f:
pickle.dump(dt, f)
return dt
def apprendre(exemples, fonc_attributs=extraction_attributs, fichier=None):
"""
Génère, à partir d'une liste d'exemples d'actions, un arbre d'apprentissage.
Cet arbre est renvoyé et stocké dans le fichier <fichier> si besoin.
"""
#Génère l'arbre d'apprentissage
data_train, data_labels = build_apprentissage(exemples, fonc_attributs)
#Apprentissage de l'arbre
dt = apprend_arbre(data_train,
data_labels,
depth=10,
feature_names=fonc_attributs.names)
#Sauvegarde de l'arbre
if fichier is not None:
with open(fichier, "wb") as f:
pickle.dump(dt, f)
return dt