khangvan
8/24/2018 - 5:03 AM
Solving Iris with Tensorflow.js and Iris JSON dataset
Solving Iris with Tensorflow.js and Iris JSON dataset
import * as tf from "@tensorflow/tfjs"
import "@tensorflow/tfjs-node"
import iris from "./iris.json"
import irisTesting from "./iris-testing.json"
// convert/setup our data
const trainingData = tf.tensor2d(iris.map(item => [
item.sepal_length, item.sepal_width, item.petal_length, item.petal_width,
]))
const outputData = tf.tensor2d(iris.map(item => [
item.species === "setosa" ? 1 : 0,
item.species === "virginica" ? 1 : 0,
item.species === "versicolor" ? 1 : 0,
]))
const testingData = tf.tensor2d(irisTesting.map(item => [
item.sepal_length, item.sepal_width, item.petal_length, item.petal_width,
]))
// build neural network
const model = tf.sequential()
model.add(tf.layers.dense({
inputShape: [4],
activation: "sigmoid",
units: 5,
}))
model.add(tf.layers.dense({
inputShape: [5],
activation: "sigmoid",
units: 3,
}))
model.add(tf.layers.dense({
activation: "sigmoid",
units: 3,
}))
model.compile({
loss: "meanSquaredError",
optimizer: tf.train.adam(.06),
})
// train/fit our network
const startTime = Date.now()
model.fit(trainingData, outputData, {epochs: 100})
.then((history) => {
// console.log(history)
model.predict(testingData).print()
})
// test network
[
{
"sepal_length": 5.4,
"sepal_width": 3.9,
"petal_length": 1.7,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 5.9,
"sepal_width": 3,
"petal_length": 5.1,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 5.7,
"sepal_width": 2.9,
"petal_length": 4.2,
"petal_width": 1.3,
"species": "versicolor"
}
][
{
"sepal_length": 5.1,
"sepal_width": 3.5,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.9,
"sepal_width": 3,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.7,
"sepal_width": 3.2,
"petal_length": 1.3,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.6,
"sepal_width": 3.1,
"petal_length": 1.5,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.6,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.6,
"sepal_width": 3.4,
"petal_length": 1.4,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.4,
"petal_length": 1.5,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.4,
"sepal_width": 2.9,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.9,
"sepal_width": 3.1,
"petal_length": 1.5,
"petal_width": 0.1,
"species": "setosa"
},
{
"sepal_length": 5.4,
"sepal_width": 3.7,
"petal_length": 1.5,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.8,
"sepal_width": 3.4,
"petal_length": 1.6,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.8,
"sepal_width": 3,
"petal_length": 1.4,
"petal_width": 0.1,
"species": "setosa"
},
{
"sepal_length": 4.3,
"sepal_width": 3,
"petal_length": 1.1,
"petal_width": 0.1,
"species": "setosa"
},
{
"sepal_length": 5.8,
"sepal_width": 4,
"petal_length": 1.2,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.7,
"sepal_width": 4.4,
"petal_length": 1.5,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 5.4,
"sepal_width": 3.9,
"petal_length": 1.3,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.5,
"petal_length": 1.4,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 5.7,
"sepal_width": 3.8,
"petal_length": 1.7,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.8,
"petal_length": 1.5,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 5.4,
"sepal_width": 3.4,
"petal_length": 1.7,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.7,
"petal_length": 1.5,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 4.6,
"sepal_width": 3.6,
"petal_length": 1,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.3,
"petal_length": 1.7,
"petal_width": 0.5,
"species": "setosa"
},
{
"sepal_length": 4.8,
"sepal_width": 3.4,
"petal_length": 1.9,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3,
"petal_length": 1.6,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.4,
"petal_length": 1.6,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 5.2,
"sepal_width": 3.5,
"petal_length": 1.5,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.2,
"sepal_width": 3.4,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.7,
"sepal_width": 3.2,
"petal_length": 1.6,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.8,
"sepal_width": 3.1,
"petal_length": 1.6,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.4,
"sepal_width": 3.4,
"petal_length": 1.5,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 5.2,
"sepal_width": 4.1,
"petal_length": 1.5,
"petal_width": 0.1,
"species": "setosa"
},
{
"sepal_length": 5.5,
"sepal_width": 4.2,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.9,
"sepal_width": 3.1,
"petal_length": 1.5,
"petal_width": 0.1,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.2,
"petal_length": 1.2,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.5,
"sepal_width": 3.5,
"petal_length": 1.3,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.9,
"sepal_width": 3.1,
"petal_length": 1.5,
"petal_width": 0.1,
"species": "setosa"
},
{
"sepal_length": 4.4,
"sepal_width": 3,
"petal_length": 1.3,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.4,
"petal_length": 1.5,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.5,
"petal_length": 1.3,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 4.5,
"sepal_width": 2.3,
"petal_length": 1.3,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 4.4,
"sepal_width": 3.2,
"petal_length": 1.3,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.5,
"petal_length": 1.6,
"petal_width": 0.6,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.8,
"petal_length": 1.9,
"petal_width": 0.4,
"species": "setosa"
},
{
"sepal_length": 4.8,
"sepal_width": 3,
"petal_length": 1.4,
"petal_width": 0.3,
"species": "setosa"
},
{
"sepal_length": 5.1,
"sepal_width": 3.8,
"petal_length": 1.6,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 4.6,
"sepal_width": 3.2,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5.3,
"sepal_width": 3.7,
"petal_length": 1.5,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 5,
"sepal_width": 3.3,
"petal_length": 1.4,
"petal_width": 0.2,
"species": "setosa"
},
{
"sepal_length": 7,
"sepal_width": 3.2,
"petal_length": 4.7,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 6.4,
"sepal_width": 3.2,
"petal_length": 4.5,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 6.9,
"sepal_width": 3.1,
"petal_length": 4.9,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 5.5,
"sepal_width": 2.3,
"petal_length": 4,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 6.5,
"sepal_width": 2.8,
"petal_length": 4.6,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 5.7,
"sepal_width": 2.8,
"petal_length": 4.5,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 6.3,
"sepal_width": 3.3,
"petal_length": 4.7,
"petal_width": 1.6,
"species": "versicolor"
},
{
"sepal_length": 4.9,
"sepal_width": 2.4,
"petal_length": 3.3,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 6.6,
"sepal_width": 2.9,
"petal_length": 4.6,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 5.2,
"sepal_width": 2.7,
"petal_length": 3.9,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 5,
"sepal_width": 2,
"petal_length": 3.5,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 5.9,
"sepal_width": 3,
"petal_length": 4.2,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 6,
"sepal_width": 2.2,
"petal_length": 4,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 6.1,
"sepal_width": 2.9,
"petal_length": 4.7,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 5.6,
"sepal_width": 2.9,
"petal_length": 3.6,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 6.7,
"sepal_width": 3.1,
"petal_length": 4.4,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 5.6,
"sepal_width": 3,
"petal_length": 4.5,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 5.8,
"sepal_width": 2.7,
"petal_length": 4.1,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 6.2,
"sepal_width": 2.2,
"petal_length": 4.5,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 5.6,
"sepal_width": 2.5,
"petal_length": 3.9,
"petal_width": 1.1,
"species": "versicolor"
},
{
"sepal_length": 5.9,
"sepal_width": 3.2,
"petal_length": 4.8,
"petal_width": 1.8,
"species": "versicolor"
},
{
"sepal_length": 6.1,
"sepal_width": 2.8,
"petal_length": 4,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 6.3,
"sepal_width": 2.5,
"petal_length": 4.9,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 6.1,
"sepal_width": 2.8,
"petal_length": 4.7,
"petal_width": 1.2,
"species": "versicolor"
},
{
"sepal_length": 6.4,
"sepal_width": 2.9,
"petal_length": 4.3,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 6.6,
"sepal_width": 3,
"petal_length": 4.4,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 6.8,
"sepal_width": 2.8,
"petal_length": 4.8,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 6.7,
"sepal_width": 3,
"petal_length": 5,
"petal_width": 1.7,
"species": "versicolor"
},
{
"sepal_length": 6,
"sepal_width": 2.9,
"petal_length": 4.5,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 5.7,
"sepal_width": 2.6,
"petal_length": 3.5,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 5.5,
"sepal_width": 2.4,
"petal_length": 3.8,
"petal_width": 1.1,
"species": "versicolor"
},
{
"sepal_length": 5.5,
"sepal_width": 2.4,
"petal_length": 3.7,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 5.8,
"sepal_width": 2.7,
"petal_length": 3.9,
"petal_width": 1.2,
"species": "versicolor"
},
{
"sepal_length": 6,
"sepal_width": 2.7,
"petal_length": 5.1,
"petal_width": 1.6,
"species": "versicolor"
},
{
"sepal_length": 5.4,
"sepal_width": 3,
"petal_length": 4.5,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 6,
"sepal_width": 3.4,
"petal_length": 4.5,
"petal_width": 1.6,
"species": "versicolor"
},
{
"sepal_length": 6.7,
"sepal_width": 3.1,
"petal_length": 4.7,
"petal_width": 1.5,
"species": "versicolor"
},
{
"sepal_length": 6.3,
"sepal_width": 2.3,
"petal_length": 4.4,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 5.6,
"sepal_width": 3,
"petal_length": 4.1,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 5.5,
"sepal_width": 2.5,
"petal_length": 4,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 5.5,
"sepal_width": 2.6,
"petal_length": 4.4,
"petal_width": 1.2,
"species": "versicolor"
},
{
"sepal_length": 6.1,
"sepal_width": 3,
"petal_length": 4.6,
"petal_width": 1.4,
"species": "versicolor"
},
{
"sepal_length": 5.8,
"sepal_width": 2.6,
"petal_length": 4,
"petal_width": 1.2,
"species": "versicolor"
},
{
"sepal_length": 5,
"sepal_width": 2.3,
"petal_length": 3.3,
"petal_width": 1,
"species": "versicolor"
},
{
"sepal_length": 5.6,
"sepal_width": 2.7,
"petal_length": 4.2,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 5.7,
"sepal_width": 3,
"petal_length": 4.2,
"petal_width": 1.2,
"species": "versicolor"
},
{
"sepal_length": 6.2,
"sepal_width": 2.9,
"petal_length": 4.3,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 5.1,
"sepal_width": 2.5,
"petal_length": 3,
"petal_width": 1.1,
"species": "versicolor"
},
{
"sepal_length": 5.7,
"sepal_width": 2.8,
"petal_length": 4.1,
"petal_width": 1.3,
"species": "versicolor"
},
{
"sepal_length": 6.3,
"sepal_width": 3.3,
"petal_length": 6,
"petal_width": 2.5,
"species": "virginica"
},
{
"sepal_length": 5.8,
"sepal_width": 2.7,
"petal_length": 5.1,
"petal_width": 1.9,
"species": "virginica"
},
{
"sepal_length": 7.1,
"sepal_width": 3,
"petal_length": 5.9,
"petal_width": 2.1,
"species": "virginica"
},
{
"sepal_length": 6.3,
"sepal_width": 2.9,
"petal_length": 5.6,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.5,
"sepal_width": 3,
"petal_length": 5.8,
"petal_width": 2.2,
"species": "virginica"
},
{
"sepal_length": 7.6,
"sepal_width": 3,
"petal_length": 6.6,
"petal_width": 2.1,
"species": "virginica"
},
{
"sepal_length": 4.9,
"sepal_width": 2.5,
"petal_length": 4.5,
"petal_width": 1.7,
"species": "virginica"
},
{
"sepal_length": 7.3,
"sepal_width": 2.9,
"petal_length": 6.3,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.7,
"sepal_width": 2.5,
"petal_length": 5.8,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 7.2,
"sepal_width": 3.6,
"petal_length": 6.1,
"petal_width": 2.5,
"species": "virginica"
},
{
"sepal_length": 6.5,
"sepal_width": 3.2,
"petal_length": 5.1,
"petal_width": 2,
"species": "virginica"
},
{
"sepal_length": 6.4,
"sepal_width": 2.7,
"petal_length": 5.3,
"petal_width": 1.9,
"species": "virginica"
},
{
"sepal_length": 6.8,
"sepal_width": 3,
"petal_length": 5.5,
"petal_width": 2.1,
"species": "virginica"
},
{
"sepal_length": 5.7,
"sepal_width": 2.5,
"petal_length": 5,
"petal_width": 2,
"species": "virginica"
},
{
"sepal_length": 5.8,
"sepal_width": 2.8,
"petal_length": 5.1,
"petal_width": 2.4,
"species": "virginica"
},
{
"sepal_length": 6.4,
"sepal_width": 3.2,
"petal_length": 5.3,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 6.5,
"sepal_width": 3,
"petal_length": 5.5,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 7.7,
"sepal_width": 3.8,
"petal_length": 6.7,
"petal_width": 2.2,
"species": "virginica"
},
{
"sepal_length": 7.7,
"sepal_width": 2.6,
"petal_length": 6.9,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 6,
"sepal_width": 2.2,
"petal_length": 5,
"petal_width": 1.5,
"species": "virginica"
},
{
"sepal_length": 6.9,
"sepal_width": 3.2,
"petal_length": 5.7,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 5.6,
"sepal_width": 2.8,
"petal_length": 4.9,
"petal_width": 2,
"species": "virginica"
},
{
"sepal_length": 7.7,
"sepal_width": 2.8,
"petal_length": 6.7,
"petal_width": 2,
"species": "virginica"
},
{
"sepal_length": 6.3,
"sepal_width": 2.7,
"petal_length": 4.9,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.7,
"sepal_width": 3.3,
"petal_length": 5.7,
"petal_width": 2.1,
"species": "virginica"
},
{
"sepal_length": 7.2,
"sepal_width": 3.2,
"petal_length": 6,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.2,
"sepal_width": 2.8,
"petal_length": 4.8,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.1,
"sepal_width": 3,
"petal_length": 4.9,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.4,
"sepal_width": 2.8,
"petal_length": 5.6,
"petal_width": 2.1,
"species": "virginica"
},
{
"sepal_length": 7.2,
"sepal_width": 3,
"petal_length": 5.8,
"petal_width": 1.6,
"species": "virginica"
},
{
"sepal_length": 7.4,
"sepal_width": 2.8,
"petal_length": 6.1,
"petal_width": 1.9,
"species": "virginica"
},
{
"sepal_length": 7.9,
"sepal_width": 3.8,
"petal_length": 6.4,
"petal_width": 2,
"species": "virginica"
},
{
"sepal_length": 6.4,
"sepal_width": 2.8,
"petal_length": 5.6,
"petal_width": 2.2,
"species": "virginica"
},
{
"sepal_length": 6.3,
"sepal_width": 2.8,
"petal_length": 5.1,
"petal_width": 1.5,
"species": "virginica"
},
{
"sepal_length": 6.1,
"sepal_width": 2.6,
"petal_length": 5.6,
"petal_width": 1.4,
"species": "virginica"
},
{
"sepal_length": 7.7,
"sepal_width": 3,
"petal_length": 6.1,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 6.3,
"sepal_width": 3.4,
"petal_length": 5.6,
"petal_width": 2.4,
"species": "virginica"
},
{
"sepal_length": 6.4,
"sepal_width": 3.1,
"petal_length": 5.5,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6,
"sepal_width": 3,
"petal_length": 4.8,
"petal_width": 1.8,
"species": "virginica"
},
{
"sepal_length": 6.9,
"sepal_width": 3.1,
"petal_length": 5.4,
"petal_width": 2.1,
"species": "virginica"
},
{
"sepal_length": 6.7,
"sepal_width": 3.1,
"petal_length": 5.6,
"petal_width": 2.4,
"species": "virginica"
},
{
"sepal_length": 6.9,
"sepal_width": 3.1,
"petal_length": 5.1,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 5.8,
"sepal_width": 2.7,
"petal_length": 5.1,
"petal_width": 1.9,
"species": "virginica"
},
{
"sepal_length": 6.8,
"sepal_width": 3.2,
"petal_length": 5.9,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 6.7,
"sepal_width": 3.3,
"petal_length": 5.7,
"petal_width": 2.5,
"species": "virginica"
},
{
"sepal_length": 6.7,
"sepal_width": 3,
"petal_length": 5.2,
"petal_width": 2.3,
"species": "virginica"
},
{
"sepal_length": 6.3,
"sepal_width": 2.5,
"petal_length": 5,
"petal_width": 1.9,
"species": "virginica"
},
{
"sepal_length": 6.5,
"sepal_width": 3,
"petal_length": 5.2,
"petal_width": 2,
"species": "virginica"
},
{
"sepal_length": 6.2,
"sepal_width": 3.4,
"petal_length": 5.4,
"petal_width": 2.3,
"species": "virginica"
}
]{
"name": "tfjs",
"version": "1.0.0",
"description": "",
"main": "index.js",
"scripts": {
"start": "parcel --target=node iris.js & nodemon dist/iris.js"
},
"keywords": [],
"author": "",
"license": "ISC",
"dependencies": {
"@tensorflow/tfjs": "^0.11.6",
"@tensorflow/tfjs-node": "^0.1.7",
"nodemon": "^1.17.5",
"parcel": "^1.9.0"
}
}