()
| 78 | } |
| 79 | |
| 80 | getTestData() { |
| 81 | const dataShape = [this.testSize, this.NUM_FEATURES]; |
| 82 | |
| 83 | const testData = Float32Array.from([].concat.apply([], this.dataset[2])); |
| 84 | const testTarget = Float32Array.from([].concat.apply([], this.dataset[3])); |
| 85 | |
| 86 | return { |
| 87 | data: tf.tensor2d(testData, dataShape), |
| 88 | target: tf.tensor1d(testTarget) |
| 89 | }; |
| 90 | } |
| 91 | } |