Skip to content

Commit e0b785c

Browse files
kkweondeep-diver
andauthored
feat: add a simple NN trainer (#8)
* feat: add a simple NN trainer * add build rule * filter data having clean column * add validation dataset * Update trainer/trainer.ts Co-authored-by: Mo Kweon <[email protected]> * Update trainer/trainer.ts Co-authored-by: Mo Kweon <[email protected]> * Update trainer/trainer.ts Co-authored-by: Mo Kweon <[email protected]> * Update trainer/package.json Co-authored-by: Mo Kweon <[email protected]> * remove unnecessary filter in tf.data Co-authored-by: Chansung Park <[email protected]>
1 parent 5a5e163 commit e0b785c

File tree

5 files changed

+1181
-1
lines changed

5 files changed

+1181
-1
lines changed

.gitignore

+2-1
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
1-
/extension/node_modules
1+
node_modules
2+
.pnpm-debug.log

trainer/package.json

+20
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
{
2+
"name": "trainer",
3+
"version": "1.0.0",
4+
"description": "",
5+
"main": "index.js",
6+
"scripts": {
7+
"build": "npx ts-node trainer.ts"
8+
},
9+
"keywords": [],
10+
"author": "",
11+
"license": "ISC",
12+
"dependencies": {
13+
"@tensorflow-models/universal-sentence-encoder": "^1.3.3",
14+
"@tensorflow/tfjs-node": "^3.18.0",
15+
"typescript": "^4.7.4"
16+
},
17+
"devDependencies": {
18+
"ts-node": "^10.8.1"
19+
}
20+
}

0 commit comments

Comments
 (0)