1- import { assert } from "chai"
2- import { MinMaxScaler , StandardScaler } from "../../src/preprocessing/scalers"
3- import { DataFrame } from '../../src/core/frame'
4- import { Series } from '../../src/core/series'
1+ import { assert } from "chai" ;
2+ import { MinMaxScaler , StandardScaler } from "../../src/preprocessing/scalers" ;
3+ import { DataFrame } from '../../src/core/frame' ;
4+ import { Series } from '../../src/core/series' ;
55
66
77describe ( "Preprocessing" , function ( ) {
88
9- describe ( "MinMaxscaler" , function ( ) {
10-
11- it ( "Standardize values in a DataFrame" , function ( ) {
12- let data = [ [ - 1 , 2 ] , [ - 0.5 , 6 ] , [ 0 , 10 ] , [ 1 , 18 ] ]
13- let scaler = new MinMaxScaler ( )
14- let fit_data = [ [ 0 , 0 ] , [ 0.25 , 0.25 ] , [ 0.5 , 0.5 ] , [ 1 , 1 ] ]
15- let transform_data = [ [ 1.5 , 0. ] ] ;
16- scaler . fit ( new DataFrame ( data ) )
17- assert . deepEqual ( scaler . transform ( new DataFrame ( data ) ) . values , fit_data )
18- assert . deepEqual ( scaler . transform ( [ [ 2 , 2 ] ] ) . values , transform_data )
19- } ) ;
20- it ( "Standardize values in a Series" , function ( ) {
21- let data = [ - 1 , 2 , - 0.5 , 60 , 101 , 18 ]
22- let scaler = new MinMaxScaler ( )
23- let result = [ 0 , 0.029411764815449715 , 0.0049019609577953815 , 0.5980392098426819 , 1 , 0.18627451360225677 ]
24- let transform_data = [ 0.029411764815449715 , 0.029411764815449715 ]
25- scaler . fit ( new Series ( data ) )
26- assert . deepEqual ( scaler . transform ( new Series ( data ) ) . values , result )
27- assert . deepEqual ( scaler . transform ( [ 2 , 2 ] ) . values , transform_data )
28- } ) ;
29- } )
30-
31- describe ( "StandardScaler" , function ( ) {
32-
33- it ( "basic test" , function ( ) {
34- let data = [ [ 0 , 0 ] , [ 0 , 0 ] , [ 1 , 1 ] , [ 1 , 1 ] ]
35-
36- let scaler = new StandardScaler ( )
37- let fit_data = [ [ - 0.9 , - 0.9 ] , [ - 0.9 , - 0.9 ] , [ 0.9 , 0.9 ] , [ 0.9 , 0.9 ] ]
38- let transform_data = [ [ 2.8 , 2.8 ] ] ;
39-
40- assert . deepEqual ( scaler . fit ( new DataFrame ( data ) ) . round ( ) . values , fit_data ) ;
41- assert . deepEqual ( scaler . transform ( [ [ 2 , 2 ] ] ) . round ( ) . values , transform_data )
42- } )
9+ describe ( "MinMaxscaler" , function ( ) {
10+
11+ it ( "Standardize values in a DataFrame" , function ( ) {
12+ let data = [ [ - 1 , 2 ] , [ - 0.5 , 6 ] , [ 0 , 10 ] , [ 1 , 18 ] ] ;
13+ let scaler = new MinMaxScaler ( ) ;
14+ let fit_data = [ [ 0 , 0 ] , [ 0.25 , 0.25 ] , [ 0.5 , 0.5 ] , [ 1 , 1 ] ] ;
15+ let transform_data = [ [ 1.5 , 0. ] ] ;
16+ scaler . fit ( new DataFrame ( data ) ) ;
17+ assert . deepEqual ( scaler . transform ( new DataFrame ( data ) ) . values , fit_data ) ;
18+ assert . deepEqual ( scaler . transform ( [ [ 2 , 2 ] ] ) . values , transform_data ) ;
19+ } ) ;
20+ it ( "Standardize values in a Series" , function ( ) {
21+ let data = [ - 1 , 2 , - 0.5 , 60 , 101 , 18 ] ;
22+ let scaler = new MinMaxScaler ( ) ;
23+ let result = [ 0 , 0.029411764815449715 , 0.0049019609577953815 , 0.5980392098426819 , 1 , 0.18627451360225677 ] ;
24+ let transform_data = [ 0.029411764815449715 , 0.029411764815449715 ] ;
25+ scaler . fit ( new Series ( data ) ) ;
26+ assert . deepEqual ( scaler . transform ( new Series ( data ) ) . values , result ) ;
27+ assert . deepEqual ( scaler . transform ( [ 2 , 2 ] ) . values , transform_data ) ;
28+ } ) ;
29+ } ) ;
30+
31+ describe ( "StandardScaler" , function ( ) {
32+
33+ it ( "basic test" , function ( ) {
34+ let data = [ [ 0 , 0 ] , [ 0 , 0 ] , [ 1 , 1 ] , [ 1 , 1 ] ] ;
35+
36+ let scaler = new StandardScaler ( ) ;
37+ let fit_data = [ [ - 1 , - 1 ] , [ - 1 , - 1 ] , [ 1 , 1 ] , [ 1 , 1 ] ] ;
38+ let transform_data = [ [ 3 , 3 ] ] ;
39+
40+ assert . deepEqual ( scaler . fit ( new DataFrame ( data ) ) . round ( ) . values , fit_data ) ;
41+ assert . deepEqual ( scaler . transform ( [ [ 2 , 2 ] ] ) . round ( ) . values , transform_data ) ;
4342 } ) ;
43+ } ) ;
4444
45- // describe("RobustScaler", function(){
46- // it("basic test", function(){
4745
48- // let data = [[3,1], [7,3], [8,4], [5,6], [12,5], [14,12], [21,23], [15,15], [18,2], [14,15]]
49- // // let data2 = [[4,100,900],[5,110,800],[21,220,890],[20,300,500]]
50- // let data2 = [0, 4, 4, 4, 7, 10, 11, 12, 14, 16, 17, 25]
46+ // describe("RobustScaler", function(){
47+ // it("basic test", function(){
5148
49+ // let data = [[3,1], [7,3], [8,4], [5,6], [12,5], [14,12], [21,23], [15,15], [18,2], [14,15]]
50+ // // let data2 = [[4,100,900],[5,110,800],[21,220,890],[20,300,500]]
51+ // let data2 = [0, 4, 4, 4, 7, 10, 11, 12, 14, 16, 17, 25]
5252
5353
54- // let scaler = new RobustScaler()
54+ // let scaler = new RobustScaler()
5555
56- // // console.log(scaler.quantile(data2,true))
56+ // // console.log(scaler.quantile(data2,true))
5757
58- // });
59- // });
60- } ) ;
58+ // });
59+ // });
60+ } ) ;
0 commit comments