feat: more tests moved over

dcrescim · dcrescim · commit 76509d8cc346 · 2022-04-30T22:34:20.000-07:00
diff --git a/src/index.ts b/src/index.ts
@@ -12,9 +12,7 @@
 * limitations under the License.
 * ==========================================================================
 */
-export { KNeighborsRegressor } from './neighbors/KNeighborsRegressor'
-export { KNeighborsClassifier } from './neighbors/KNeighborsClassifier'
-export { makeRegression, makeLowRankMatrix } from './datasets/makeRegression'
+
 export {
   LinearRegression,
   LinearRegressionParams
@@ -71,6 +69,9 @@ export {
   VotingClassifier,
   VotingClassifierParams
 } from './ensemble/VotingClassifier'
+export { KNeighborsRegressor } from './neighbors/KNeighborsRegressor'
+export { KNeighborsClassifier } from './neighbors/KNeighborsClassifier'
+
 export { LinearSVC, LinearSVCParams } from './svm/LinearSVC'
 export { LinearSVR, LinearSVRParams } from './svm/LinearSVR'
 
@@ -87,4 +88,6 @@ export {
 } from './tree/DecisionTree'
 export { trainTestSplit } from './model_selection/trainTestSplit'
 export { KFold } from './model_selection/KFold'
+export { crossValScore } from './model_selection/crossValScore'
+export { makeRegression, makeLowRankMatrix } from './datasets/makeRegression'
 export { setBackend, getBackend } from './tf-singleton'
diff --git a/src/neighbors/KNeighborsRegressor.test.ts b/src/neighbors/KNeighborsRegressor.test.ts
@@ -13,12 +13,11 @@
 * ==========================================================================
 */
 
-import { KNeighborsRegressor, setBackend } from '../index'
+import { KNeighborsRegressor, setBackend, KFold } from '../index'
+import { crossValScore } from '../model_selection/crossValScore'
 import { KNeighborsParams } from './KNeighborsBase'
 import { dataUrls } from '../datasets/datasets'
 import { arrayEqual } from '../utils'
-import { crossValScore } from '../model_selection/crossValScore'
-import { KFold } from '../model_selection/KFold'
 import { negMeanSquaredError } from '../model_selection/scorers'
 import '../jestTensorMatchers'
 import * as dfd from 'danfojs-node'
@@ -45,7 +44,7 @@ function testWithDataset(
       const X = Xy.slice([0, 0], [nSamples, nFeatures])
       const y = Xy.slice([0, nFeatures]).reshape([nSamples]) as Tensor1D
 
-      const scores = await crossValScore(
+      const scores = await (crossValScore as any)(
         new KNeighborsRegressor(params),
         X,
         y,
