tf-simple
v0.1.5
Published
tf-simple allows for easy neural network creation and training in tensorflow.js
Downloads
2
Maintainers
Readme
Example Usage
const train = require('tf-simple').train;
const data = [
{inputs: [0,0, 'yeet', 'teey'], outputs: [0]},
{inputs: [0,1, 'yeet', 'teey'], outputs: [1]},
{inputs: [1,0, 'yeet', 'teey'], outputs: [1]},
{inputs: [1,1, 'yeet', 'teey'], outputs: [0]},
];
const arch = [
{units: 12, activation: 'relu'},
{units: 1, activation: 'sigmoid'}
]
(async()=>{
await train({
data,
arch,
useValLoss: false,
desiredLoss: .00001,
validation: .3,
shuffle: false,
verbose: 2
});
})();