-
Notifications
You must be signed in to change notification settings - Fork 0
/
tfjs-train-template.html.template
51 lines (47 loc) · 1.5 KB
/
tfjs-train-template.html.template
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
51
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>TFJS Training Template</title>
</head>
<body>
<script src="https://cdnjs.cloudflare.com/ajax/libs/tensorflow/1.3.2/tf.min.js"></script>
<script>
function createModel() {
const model = null;
return model;
}
function createData() {
const data = {x: null, y: null};
return data;
}
async function train(model, data, epochs = 500) {
const history = await model.fit(data.x, data.y, {
epochs: epochs,
callbacks: {
onEpochEnd: async (epoch, logs) => {
if (epoch % 10 == 0) {
console.debug(`Loss is ${logs.loss} after ${epoch} epochs`);
// console.log(logs);
}
}
}
});
return history;
}
async function main() {
const model = createModel();
const data = createData();
const history = await train(model, data);
console.debug(history);
console.log("Done training!")
return model;
}
main().then(model => {
// model.predict(tf.tensor2d([10], [1, 1])).print()
})
</script>
</body>
</html>