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

implement Convolution1D layer, with tests

Browse Source
This commit is contained in:
Leon Chen
2016-08-26 20:25:21 -04:00
parent 263242e6e4
commit ef6f18bd0a
11 changed files with 589 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/convolutional/Convolution2D.js
+2 -2
View File
@@ -63,7 +63,7 @@ describe('convolutional layer: Convolution2D', function () {
const key = `convolutional.Convolution2D.${i}`
const [inputRows, inputCols, inputChannels] = inputShape
const [nbFilter, nbRow, nbCol] = kernelShape
const title = `[${key}] [CPU] test 1: ${nbFilter} ${nbRow}x${nbCol} filters on ${inputRows}x${inputCols}x${inputChannels} input, activation='${attrs.activation}', border_mode='${attrs.borderMode}', subsample=${attrs.subsample}, dim_ordering='${attrs.dimOrdering}', bias=${attrs.bias}`
const title = `[${key}] [CPU] test: ${nbFilter} ${nbRow}x${nbCol} filters on ${inputRows}x${inputCols}x${inputChannels} input, activation='${attrs.activation}', border_mode='${attrs.borderMode}', subsample=${attrs.subsample}, dim_ordering='${attrs.dimOrdering}', bias=${attrs.bias}`
it(title, function () {
console.log(`\n%c${title}`, styles.h3)
@@ -97,7 +97,7 @@ describe('convolutional layer: Convolution2D', function () {
const key = `convolutional.Convolution2D.${i}`
const [inputRows, inputCols, inputChannels] = inputShape
const [nbFilter, nbRow, nbCol] = kernelShape
const title = `[${key}] [GPU] test 1: ${nbFilter} ${nbRow}x${nbCol} filters on ${inputRows}x${inputCols}x${inputChannels} input, activation='${attrs.activation}', border_mode='${attrs.borderMode}', subsample=${attrs.subsample}, dim_ordering='${attrs.dimOrdering}', bias=${attrs.bias}`
const title = `[${key}] [GPU] test: ${nbFilter} ${nbRow}x${nbCol} filters on ${inputRows}x${inputCols}x${inputChannels} input, activation='${attrs.activation}', border_mode='${attrs.borderMode}', subsample=${attrs.subsample}, dim_ordering='${attrs.dimOrdering}', bias=${attrs.bias}`
it(title, function () {
console.log(`\n%c${title}`, styles.h3)