-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.js
More file actions
50 lines (50 loc) · 1.54 KB
/
Copy pathmain.js
File metadata and controls
50 lines (50 loc) · 1.54 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
var json;
var dataset = [];
var net = new brain.recurrent.LSTM();
if (json) {
net.fromJSON(json);
}
var iter = 0;
var load = function (d) {
d = d.toString().toLowerCase();
console.log("Finding sentences...");
dataset = d.toString().split("\t").join(" ").toString().split("\n");
console.log("Loaded dataset...");
console.log(dataset);
if (!brain) {
console.log("Missing dependencie 'brain.js'");
return;
}
document.getElementById("train").addEventListener("click", function () {
console.log("Training...");
var x = iter;
var y = iter + parseInt(window.prompt("Training Size:"));
iter = y;
net.train(dataset.slice(0,y), {
iterations: 10000,
errorThresh: 0.001,
log: true,
logPeriod: 1,
learningRate: 0.2,
momentum: 1,
callback: null,
callbackPeriod: 10,
timeout: 1000*60,
});
console.log(net.toFunction());
console.log(net.toJSON());
document.getElementById("trans").addEventListener("click", function () {
var input = document.getElementById("tr").value;
var output = net.run(input+" ");
document.getElementById("out").innerHTML = "<span>"+input+"</span> = "+output;
console.log(output);
});
});
};
var xhr = new XMLHttpRequest();
xhr.open("GET", "jpn.txt");
xhr.addEventListener("load", function (x) {
console.log("Loaded file...");
load(this.responseText);
});
xhr.send();