-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathbot.rl.R
45 lines (39 loc) · 1.91 KB
/
bot.rl.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# Chargement de la biblioth?que
library (e1071)
# Chargement des donn?es
awele.data = read.table ("awele.data", sep = ",", header = T)
####################
# Premi?re version #
####################
# On s?lectionne toutes les observations correspondant aux coups jou?s par le gagnant de la partie
# On construit un mod?le de Naive Bayes ? partir des 12 premi?res variables de ces observations
# On essaye de pr?dire la 13e variable (coup jou?)
# Fonction de construction du mod?les
RL.create.model = function (dataset)
{
# On s?lectionne les instances qui correspondent aux coups jou?s par le vainqueur des affrontements
selection = awele.data [dataset [, 14] == "G", ]
# Et on construit un mod?le de classification avec l'algorithme Naive bayes
model = LR(selection [, 1:12], selection [, 13])
return (model)
}
# Construction du mod?le
RL.model = RL.create.model (awele.data)
# Fonction d'?valuation de la meilleure solution selon l'?tat du plateau de jeu et du mod?le
RL.exec = function (awele, model)
{
# On r?cup?re l'?tat du plateau de jeu (sous la forme d'une matrice plut?t que d'un vecteur)
g = graines.matrix (awele)
# On modifie les noms des colonnes pour correspondre aux noms dans l'ensemble d'apprentissage
colnames (g) = c (paste ("J", 1:6, sep = ""), paste ("A", 1:6, sep = ""))
# On applique le mod?le et on retourne les pr?dictions (sous la forme de degr?s d'appartenance aux classes)
prediction = predict (model, data.frame(g),type = "raw")
# On crée une nouvelle dataframe pour le résultat, tout à 0, et on y colle les noms des colonnes
ret = data.frame(matrix(data=0, ncol=6, nrow=1))
colnames(ret) = levels(prediction)
# on set la colonne retournée par prédict ) 1.
ret[c(prediction[1])] = 1
return (ret)
}
# Fonction d'?valuation de la meilleure solution selon l'?tat du plateau de jeu (en utilisant la variable globale RL.model)
RL = function (awele) return (RL.exec (awele, RL.model))