Brain.js to detect bad words

[Bhargav230m]

Hey there i wanted to train a model to detect badwords, i have tried downloading brain.js to train a model. I have my badwords dataset in json file. I wanted to know if i am going the right track .

import brain from "brain.js";
import fs from "fs";

const a = [];
try {
  const data = fs.readFileSync("BadWord.json", "utf8");
  const parsedData = JSON.parse(data);
  parsedData.known_links.forEach((word) => {
    a.push({ input: word, output: true });
  });

  const net = new brain.recurrent.RNN();
  net.train(a, {
    log: true,
    iterations: 5000,
    learningRate: 0.0001
  });

  const netData = net.toJSON();
  fs.writeFileSync("model.json", netData);

  const b = net.run("goodWord");
  console.log("Result of running model " + b);
} catch (err) {
  console.error(err);
}

[Bhargav230m]

I had tried learning rate 0.001 earlier but that resulted in Infinity Training error.

Currently my training error is at 0.03

[robertleeplummerjr]

You should really use a LSTM or a GRU for that type of problem.