Help ! :p

prenom_custom.csv

StupidAI.js

const brain = require('brain.js');
const fs = require('fs')
var validator = require('validator');
var csv = require('csv-parser')

const net = new brain.NeuralNetwork();

let trainingData = [];

fs.createReadStream('data/prenom_custom.csv')
    .pipe(csv())
    .on('data', (data) => {

        if (validator.isAlpha(data.prenom, 'fr-FR') == true) {
            if (data.genre == 2) {
                trainingData.push({ input: data.prenom, output: { femme: 1, homme: 0 } })
            }
            if (data.genre == 1) {
                trainingData.push({ input: data.prenom, output: { femme: 0, homme: 1 } })
            }
        }

    })
    .on('end', () => {

        let serializedData = serialize(trainingData);

        const config = {
            log: true,
            logPeriod: 100
        };

        net.train(serializedData, config);

        console.log(net.run(encodeForTest("Noémie", 12)));
        console.log(net.run(encodeForTest("Julie", 12)));
        console.log(net.run(encodeForTest("Claire", 12)));
        console.log(net.run(encodeForTest("Amandine", 12)));
        console.log(net.run(encodeForTest("Laurence", 12)));

        console.log('----------')

        console.log(net.run(encodeForTest("Lucas", 12)));
        console.log(net.run(encodeForTest("Maxime", 12)));
        console.log(net.run(encodeForTest("Arthur", 12)));
        console.log(net.run(encodeForTest("Romain", 12)));
        console.log(net.run(encodeForTest("Carlito", 12)));

    });



function fixLengths(data) {
    let maxLengthInput = -1;
    for (let i = 0; i < data.length; i++) {
        if (data[i].input.length > maxLengthInput) {
            maxLengthInput = data[i].input.length;
        }
    }

    for (let i = 0; i < data.length; i++) {
        while (data[i].input.length < maxLengthInput) {
            data[i].input.push(0);
        }
    }

    return data;
}


function encodeData(data) {
    return data.map(d => {
        return {
            input: encode(d.input),
            output: d.output
        }
    });
}


function serialize(data) {
    return fixLengths(encodeData(data));
};


function encodeForTest(d, nombre) {
    const newArr = [];
    d.split("").forEach(c => {
        newArr.push((c.charCodeAt(0) / 255))
    });
    while (newArr.length < nombre) {
        newArr.push(0);
    }
    return newArr;
}


function encode(d) {
    const newArr = [];
    d.split("").forEach(c => {
        newArr.push((c.charCodeAt(0) / 255))
    });
    while (newArr.length < 7) {
        newArr.push(0);
    }
    return newArr;
}