05-learn-phone-price.js

const { loadOrTrain } = require("./utils")
const brain = require("brain.js")

const trainingData = require("./selectra-phone-dataset-numbers-only-normalized.json");

const net = new brain.NeuralNetwork({ hiddenLayers: [3] });

const stats = net.train(trainingData);
// const stats = loadOrTrain(net, trainingData);

console.log(stats);

let sampleInput = {"Processeur":0.5,"Mémoire vive (RAM)":0.5,"Mémoire interne":0.25,"Poids":0.7178217821782178,"Appareil photo frontal":0.3125,"Autonomie":0.6944444444444444,"Écran":0.43243243243243246}

console.log(net.run(sampleInput));