0.1.5 • Published 4 years ago

tf-simple v0.1.5

Weekly downloads
1
License
MIT
Repository
-
Last release
4 years ago

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
    });
    
})();