wassname/keras-js
1
0
Fork 0
mirror of https://github.com/wassname/keras-js.git synced 2026-09-11 12:20:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

move all layer params into attrs object

Browse Source
This commit is contained in:
Leon Chen
2016-09-18 23:20:59 -04:00
parent b1af2a7751
commit fd20f1b7b1
47 changed files with 285 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/core/Dropout.js
+2 -2
View File
@@ -15,12 +15,12 @@ describe('core layer: Dropout', function () {
it('[core.Dropout.0] should just pass through tensor during test time', function () {
const key = 'core.Dropout.0'
console.log(`\n%c[${key}] should pass through`, styles.h3)
let testLayer1 = new layers.Dense(2)
let testLayer1 = new layers.Dense({ outputDim: 2 })
testLayer1.setWeights(TEST_DATA[key].weights.map(w => new KerasJS.Tensor(w.data, w.shape)))
let t = new KerasJS.Tensor(TEST_DATA[key].input.data, TEST_DATA[key].input.shape)
t = testLayer1.call(t)
console.log('%cin', styles.h4, stringifyCondensed(t.tensor))
let testLayer2 = new layers.Dropout(0.5)
let testLayer2 = new layers.Dropout({ p: 0.5 })
const startTime = performance.now()
t = testLayer2.call(t)
const endTime = performance.now()