Skip to content

Latest commit

 

History

History
 
 

iris

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

Use Flux to do Logistic Regression on the Iris dataset

This is a very simple model, with a single layer that outputs to softmax.

Logistic regression can basically be thought of as a single layer neural network.

Data Source

The data source is Fisher's classic dataset, retrieved from the UCI Machine Learning Repository.

Usage

cd into model-zoo/other/iris, start the Julia REPL and instantiate the environment:

julia> using Pkg; Pkg.activate("."); Pkg.instantiate()

Then train and evaluate the model:

julia> include("iris.jl")
Starting training.

Accuracy: 0.94

Confusion Matrix:

3×3 Array{Int64,2}:
 16   0   0
  0  16   1
  0   2  15

julia>