Come tracciare un oggetto svm in r (con esempio)

È possibile utilizzare la seguente sintassi di base per tracciare un oggetto Support Vector Machine (SVM) in R:

 library (e1071)

plot(svm_model, df)

In questo esempio, df è il nome del frame di dati e svm_model è un vettore di supporto adattato alla macchina utilizzando la funzione svm() .

L’esempio seguente mostra come utilizzare questa sintassi nella pratica.

Esempio: come tracciare un oggetto SVM in R

Supponiamo di avere il seguente frame di dati in R che contiene informazioni su vari giocatori di basket:

 #create data frame
df <- data. frame (points = c(4, 5, 5, 7, 8, 12, 15, 22, 25, 29),
                 assists = c(3, 4, 6, 8, 5, 6, 5, 6, 8, 12),
                 good = factor(c(0, 0, 0, 1, 0, 1, 0, 1, 1, 1)))

#view data frame
df

   points assists good
1 4 3 0
2 5 4 0
3 5 6 0
4 7 8 1
5 8 5 0
6 12 6 1
7 15 5 0
8 22 6 1
9 25 8 1
10 29 12 1

Diciamo che vogliamo creare una macchina vettoriale di supporto che utilizzi punti variabili e aiuti a prevedere se un giocatore è bravo o meno (0 = no, 1 = sì).

Possiamo utilizzare il seguente codice per adattare la macchina vettoriale di supporto e quindi tracciare i risultati:

 library (e1071)

#fit support vector machine
model = svm(good ~ points + assists, data = df)

#plot support vector machine
plot(model, df) 

Il grafico mostra i valori della variabile assist sull’asse x, i valori della variabile punti sull’asse y e utilizza due colori diversi per visualizzare se si prevede che un giocatore sarà bravo (rosso) o non (giallo).

Tieni presente che puoi utilizzare l’argomento color.palette nella funzione plot() per utilizzare una tavolozza di colori diversa per la trama.

Ad esempio, potremmo scegliere di utilizzare la tavolozza dei colori heat.colors :

 library (e1071)

#fit support vector machine
model = svm(good ~ points + assists, data = df)

#plot support vector machine using different color palette
plot(model, df, color. palette = heat. colors ) 

Altre scelte popolari per l’argomento color.palette includono:

• Arcobaleno
• colori.del.terreno
• topo.colors

Ogni tavolozza di colori produrrà colori diversi per la trama.

Risorse addizionali

I seguenti tutorial spiegano come eseguire altre attività comuni in R:

Come filtrare valori univoci utilizzando dplyr
Come filtrare in base a più condizioni utilizzando dplyr
Come contare il numero di occorrenze nelle colonne in R