Fix constructor of APL layer (#920)

Author: ishii-norimi

lib/model/nns/layer/apl.js

@@ -10,18 +10,18 @@ export default class AdaptivePiecewiseLinearLayer extends Layer {
 	 * @param {number | number[]} [config.a] Variables control the slopes of the linear segments
 	 * @param {number | number[]} [config.b] Variables determine the locations of the hinges
 	 */
-	constructor({ s = 2, a = 0.1, b = 0, ...rest }) {
+	constructor({ s = 2, a = null, b = 0, ...rest }) {
 		super(rest)
 		this._s = s
 		if (Array.isArray(a)) {
 			this._a = a
 		} else {
 			this._a = []
 			for (let k = 0; k < s; k++) {
-				this._a[k] = Math.random()
+				this._a[k] = a ?? Math.random()
 			}
 		}
-		this._b = Array.isArray(b) ? b : Array(s).fill(0)
+		this._b = Array.isArray(b) ? b : Array(s).fill(b)
 		this._l2_decay = 0.001
 	}
 

tests/lib/model/nns/layer/apl.test.js

@@ -1,4 +1,4 @@
-import { jest } from '@jest/globals'
+import { expect, jest, test } from '@jest/globals'
 jest.retryTimes(3)
 
 import NeuralNetwork from '../../../../../lib/model/neuralnetwork.js'
@@ -8,9 +8,16 @@ import Tensor from '../../../../../lib/util/tensor.js'
 import APLLayer from '../../../../../lib/model/nns/layer/apl.js'
 
 describe('layer', () => {
-	test('construct', () => {
-		const layer = new APLLayer({})
-		expect(layer).toBeDefined()
+	describe('construct', () => {
+		test('default', () => {
+			const layer = new APLLayer({})
+			expect(layer).toBeDefined()
+		})
+
+		test('number', () => {
+			const layer = new APLLayer({ a: 2, b: 3 })
+			expect(layer).toBeDefined()
+		})
 	})
 
 	describe('calc', () => {