Skip to content

Instantly share code, notes, and snippets.

@kmaher9
Last active September 5, 2018 20:50
Show Gist options
  • Save kmaher9/f7de043b436c28abe9d9c3b69143a1f4 to your computer and use it in GitHub Desktop.
Save kmaher9/f7de043b436c28abe9d9c3b69143a1f4 to your computer and use it in GitHub Desktop.
var fs = require('fs') // to get directory listings, and read files - npm install fs
var brain = require('brain.js') // contains core functions - npm install brain.js
var businessPath = "path/to/business/dataset" // location of 100 training files
var techPath = "path/to/tech/dataset" // second class 100 training files
var testFile = "path/to/test/file.txt" // file which will be used to test accuracy - belongs to tech
var inputs = [] // will be used to store all training data prior to training
var errorThreshold = 0.085 // the minimum error margin needed to class the nn as trained
var iterations = 100 // the maximum number of training iterations to meet, if the errorThreshold is not met
var net = new brain.recurrent.LSTM() // the instance of the LSTM model I will be using
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment