wassname/keras-js
1
0
Fork 0
mirror of https://github.com/wassname/keras-js.git synced 2026-09-10 12:15:12 +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/RepeatVector.js
+1 -1
View File
@@ -15,7 +15,7 @@ describe('core layer: RepeatVector', function () {
it('[core.RepeatVector.0] should be able to go from shape [6] -> [7, 6]', function () {
const key = 'core.RepeatVector.0'
console.log(`\n%c[${key}] repeat vector, shape [6] -> [7, 6]`, styles.h3)
let testLayer = new layers.RepeatVector(7)
let testLayer = new layers.RepeatVector({ n: 7 })
let t = new KerasJS.Tensor(TEST_DATA[key].input.data, TEST_DATA[key].input.shape)
console.log('%cin', styles.h4, stringifyCondensed(t.tensor))
const startTime = performance.now()