¿Hay alguna manera de usar brain.js para dar una serie de números y obtener una salida categórica (preferiblemente acompañada de un valor de confiabilidad)?

Supongamos que este código

 const brain = require('brain.js'); const network = new brain.NeuralNetwork(); //IS THIS THE ONE? network.train([ // [word count, characters, special characters] => [rating] { input: [0.10, 0.20, 0.40], output: ['hard'] }, // 1 input { input: [0.10, 0.50, 0.50], output: ['intermediate'] }, // 2 input { input: [0.10, 0.40, 0.60], output: ['simple'] }, // 3 input { input: [0.10, 0.40, 0.50], output: ['hard'] }, // 4 input { input: [0.10, 0.30, 0.30], output: ['simple'] }, // 5 input { input: [0.10, 0.70, 0.30], output: ['hard'] } // 6 input ]); //Evaluate to get a difficulty rating const output = network.run([0.10, 0.50, 0.60]); console.log(output);

El resultado deseado sería una de las tres clasificaciones con un valor de confiabilidad (quizás un porcentaje de certeza).