Skip to content

Instantly share code, notes, and snippets.

@JoosKorstanje

JoosKorstanje/julia_ml_004.jl

Created October 4, 2020 08:44
Show Gist options
  • Save JoosKorstanje/89894215f34b9104f901443124daa6cd to your computer and use it in GitHub Desktop.
Save JoosKorstanje/89894215f34b9104f901443124daa6cd to your computer and use it in GitHub Desktop.
Download ZIP
julia_ml_004.jl
Raw
julia_ml_004.jl
import Pkg; Pkg.add("StatsModels")
import Pkg; Pkg.add("GLM")
using DataFrames, GLM
fm_setosa = @formula(Setosa ~ sepallength + sepalwidth + petallength + petalwidth)
lm_setosa = glm(fm_setosa, train, Binomial(), LogitLink())
pred_setosa = predict(lm_setosa, test)
fm_virginica = @formula(Virginica ~ sepallength + sepalwidth + petallength + petalwidth)
lm_virginica = glm(fm_virginica, train, Binomial(), LogitLink())
pred_virginica = predict(lm_virginica, test)
fm_versicolor = @formula(Versicolor ~ sepallength + sepalwidth + petallength + petalwidth)
lm_versicolor = glm(fm_versicolor, train, Binomial(), LogitLink())
pred_versicolor = predict(lm_versicolor, test)
preds = hcat(pred_setosa, pred_virginica, pred_versicolor)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment