Como plotar um objeto svm em r (com exemplo)

Você pode usar a seguinte sintaxe básica para plotar um objeto de máquina de vetores de suporte (SVM) em R:

 library (e1071)

plot(svm_model, df)

Neste exemplo, df é o nome do quadro de dados e svm_model é um ajuste de máquina de vetores de suporte usando a função svm() .

O exemplo a seguir mostra como usar essa sintaxe na prática.

Exemplo: como plotar um objeto SVM em R

Suponha que temos o seguinte quadro de dados em R que contém informações sobre vários jogadores de basquete:

 #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

Digamos que queremos criar uma máquina de vetores de suporte que use pontos variáveis e ajude a prever se um jogador é bom ou não (0 = não, 1 = sim).

Podemos usar o seguinte código para ajustar a máquina de vetores de suporte e então plotar os resultados:

 library (e1071)

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

#plot support vector machine
plot(model, df) 

O gráfico exibe os valores da variável de assistência no eixo x, os valores da variável de pontos no eixo y e usa duas cores diferentes para exibir se um jogador está previsto para ser bom (vermelho) ou não (amarelo).

Observe que você pode usar o argumento color.palette na função plot() para usar uma paleta de cores diferente para o gráfico.

Por exemplo, poderíamos optar por usar a paleta de cores 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 ) 

Outras opções populares para o argumento color.palette incluem:

• Arco-íris
• terreno.colors
• topo.colors

Cada paleta de cores produzirá cores diferentes para o gráfico.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns em R:

Como filtrar valores únicos usando dplyr
Como filtrar por múltiplas condições usando dplyr
Como contar o número de ocorrências em colunas em R