solid-droid / SimpleML

Machine Learning made Simple

SimpleML

Machine Learning made Simple
Wrapper for tensorflowjs

Create -> Train -> Predict

Goals

• simplify tensorflow js
• Distributed ML over webRTC.
• Easy to use

How To Use

HTML

<script src="./ts.min.js"></script>
<script src="./simpleML.js"></script>

Javascript

const nn = new simpleML.network();

//create inbuilt sequential network
nn.input(2)
  .layer(2, 'sigmoid')
  .layer(1, 'sigmoid')

//dataset
const xor = [
    { input:[0,0], output:[0]  },
    { input:[0,1], output:[1]  },
    { input:[1,0], output:[1]  },
    { input:[1,1], output:[0]  },  
]

//inbuilt compile and fit method using backpropagation
nn.train(xor, {
  epochs: 1500,
  learningRate: 0.01,
  optimizer: 'adam',
  onEpochEnd : (epoch, logs) => console.log(epoch, logs),
  onTrainEnd : () => predict()
});

//predict
const predict = () => {
  const res = nn.predict([[0,0],[1,1],[0,1],[1,0]], {
    precision: 3,
    round: true
  });

  console.log(res)
  //[0, 0, 1, 1]
};