Small improve

Author: ishii-norimi

lib/util/math.js

@@ -1500,14 +1500,15 @@ export class Matrix {
 	/**
 	 * Shuffle values along the axis.
 	 * @param {number} [axis=0]
+	 * @returns {number[]} Original index.
 	 */
 	shuffle(axis = 0) {
 		const idx = []
 		for (let i = 0; i < this._size[axis]; i++) {
 			idx[i] = i
 		}
 		for (let i = idx.length - 1; i > 0; i--) {
-			let r = Math.floor(Math.random() * (i + 1))
+			const r = Math.floor(Math.random() * (i + 1))
 			;[idx[i], idx[r]] = [idx[r], idx[i]]
 		}
 
@@ -1516,6 +1517,7 @@ export class Matrix {
 		} else if (axis === 1) {
 			this._value = this.col(idx)._value
 		}
+		return idx
 	}
 
 	/**

tests/lib/model/confidence_weighted.test.js

@@ -1,3 +1,6 @@
+import { jest } from '@jest/globals'
+jest.retryTimes(3)
+
 import { Matrix } from '../../../lib/util/math.js'
 import { ConfidenceWeighted, SoftConfidenceWeighted } from '../../../lib/model/confidence_weighted.js'