1- import { assert } from "chai"
2- import { LabelEncoder , OneHotEncoder } from "../../src/preprocessing/encodings"
3- import { Series } from "../../src/core/series"
1+ import { assert } from "chai" ;
2+ import { LabelEncoder , OneHotEncoder } from "../../src/preprocessing/encodings" ;
3+ import { Series } from "../../src/core/series" ;
44
55describe ( "Encodings" , function ( ) {
66
7- describe ( "LabelEncoder" , function ( ) {
7+ describe ( "LabelEncoder" , function ( ) {
88
9- it ( "test the label encoding on array" , function ( ) {
10- let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ]
11- let encode = new LabelEncoder ( )
12- let fit_data = [
13- 0 , 1 , 2 , 0 ,
14- 1 , 2 , 2 , 1
15- ]
16- assert . deepEqual ( encode . fit ( data ) . values , fit_data )
17- assert . deepEqual ( encode . transform ( [ "dog" , "man" ] ) . values , [ 0 , 2 ] )
18- } ) ;
19- it ( "test the label encoding on Series" , function ( ) {
20- let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ]
21- let series = new Series ( data )
22- let encode = new LabelEncoder ( )
23- let fit_data = [
24- 0 , 1 , 2 , 0 ,
25- 1 , 2 , 2 , 1
26- ]
27- assert . deepEqual ( encode . fit ( series ) . values , fit_data )
28- assert . deepEqual ( encode . transform ( [ "dog" , "man" ] ) . values , [ 0 , 2 ] )
29- } ) ;
30- it ( "label encoding directly from a Series" , function ( ) {
31- let data = new Series ( [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ] )
32- let to_label_encode = new Series ( [ "dog" , "man" ] )
33- let encode = new LabelEncoder ( )
34- let fit_data = [
35- 0 , 1 , 2 , 0 ,
36- 1 , 2 , 2 , 1
37- ]
38- assert . deepEqual ( encode . fit ( data ) . values , fit_data )
39- assert . deepEqual ( encode . transform ( to_label_encode ) . values , [ 0 , 2 ] )
40- } ) ;
41- // it("Label encoding on Series", function () {
42- // let data = ["dog", "cat", "man", "dog", "cat", "man", "man", "cat"]
43- // let series = new Series(data)
44- // let encode = new LabelEncoder()
45- // let fit_data = [
46- // 0, 1, 2, 0,
47- // 1, 2, 2, 1
48- // ]
49- // assert.deepEqual(encode.fit(series).values, fit_data)
50- // assert.deepEqual(encode.transform(["dog", "man"]).values, [0, 2])
51- // });
9+ it ( "test the label encoding on array" , function ( ) {
10+ let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ] ;
11+ let encode = new LabelEncoder ( ) ;
12+ let fit_data = [
13+ 0 , 1 , 2 , 0 ,
14+ 1 , 2 , 2 , 1
15+ ] ;
16+ assert . deepEqual ( encode . fit ( data ) . values , fit_data ) ;
17+ assert . deepEqual ( encode . transform ( [ "dog" , "man" ] ) . values , [ 0 , 2 ] ) ;
18+ } ) ;
19+ it ( "test the label encoding on Series" , function ( ) {
20+ let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ] ;
21+ let series = new Series ( data ) ;
22+ let encode = new LabelEncoder ( ) ;
23+ let fit_data = [
24+ 0 , 1 , 2 , 0 ,
25+ 1 , 2 , 2 , 1
26+ ] ;
27+ assert . deepEqual ( encode . fit ( series ) . values , fit_data ) ;
28+ assert . deepEqual ( encode . transform ( [ "dog" , "man" ] ) . values , [ 0 , 2 ] ) ;
29+ } ) ;
30+ it ( "label encoding directly from a Series" , function ( ) {
31+ let data = new Series ( [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ] ) ;
32+ let to_label_encode = new Series ( [ "dog" , "man" ] ) ;
33+ let encode = new LabelEncoder ( ) ;
34+ let fit_data = [
35+ 0 , 1 , 2 , 0 ,
36+ 1 , 2 , 2 , 1
37+ ] ;
38+ assert . deepEqual ( encode . fit ( data ) . values , fit_data ) ;
39+ assert . deepEqual ( encode . transform ( to_label_encode ) . values , [ 0 , 2 ] ) ;
40+ } ) ;
41+ // it("Label encoding on Series", function () {
42+ // let data = ["dog", "cat", "man", "dog", "cat", "man", "man", "cat"]
43+ // let series = new Series(data)
44+ // let encode = new LabelEncoder()
45+ // let fit_data = [
46+ // 0, 1, 2, 0,
47+ // 1, 2, 2, 1
48+ // ]
49+ // assert.deepEqual(encode.fit(series).values, fit_data)
50+ // assert.deepEqual(encode.transform(["dog", "man"]).values, [0, 2])
51+ // });
5252
53- } )
53+ } ) ;
5454
55- describe ( "OneHotEncoder" , function ( ) {
55+ describe ( "OneHotEncoder" , function ( ) {
5656
57- it ( "test onehotencoding on array" , function ( ) {
58- let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ]
59- let encode = new OneHotEncoder ( )
60- let fit_data = [
61- [ 1 , 0 , 0 ] ,
62- [ 0 , 1 , 0 ] ,
63- [ 0 , 0 , 1 ] ,
64- [ 1 , 0 , 0 ] ,
65- [ 0 , 1 , 0 ] ,
66- [ 0 , 0 , 1 ] ,
67- [ 0 , 0 , 1 ] ,
68- [ 0 , 1 , 0 ]
69- ]
70- let transform_data = [ [ 0 , 0 , 1 ] , [ 0 , 1 , 0 ] ]
57+ it ( "test onehotencoding on array" , function ( ) {
58+ let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ] ;
59+ let encode = new OneHotEncoder ( ) ;
60+ let fit_data = [
61+ [ 1 , 0 , 0 ] ,
62+ [ 0 , 1 , 0 ] ,
63+ [ 0 , 0 , 1 ] ,
64+ [ 1 , 0 , 0 ] ,
65+ [ 0 , 1 , 0 ] ,
66+ [ 0 , 0 , 1 ] ,
67+ [ 0 , 0 , 1 ] ,
68+ [ 0 , 1 , 0 ]
69+ ] ;
70+ let transform_data = [ [ 0 , 0 , 1 ] , [ 0 , 1 , 0 ] ] ;
7171
72- assert . deepEqual ( encode . fit ( data ) . values , fit_data ) ;
73- assert . deepEqual ( encode . transform ( [ "man" , "cat" ] ) . values , transform_data )
74- } )
75- it ( "test onehotencoding on Series" , function ( ) {
76- let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ]
77- let series = new Series ( data )
78- let encode = new OneHotEncoder ( )
79- let fit_data = [
80- [ 1 , 0 , 0 ] ,
81- [ 0 , 1 , 0 ] ,
82- [ 0 , 0 , 1 ] ,
83- [ 1 , 0 , 0 ] ,
84- [ 0 , 1 , 0 ] ,
85- [ 0 , 0 , 1 ] ,
86- [ 0 , 0 , 1 ] ,
87- [ 0 , 1 , 0 ]
88- ]
89- let transform_data = [ [ 0 , 0 , 1 ] , [ 0 , 1 , 0 ] ]
72+ assert . deepEqual ( encode . fit ( data ) . values , fit_data ) ;
73+ assert . deepEqual ( encode . transform ( [ "man" , "cat" ] ) . values , transform_data ) ;
74+ } ) ;
75+ it ( "test onehotencoding on Series" , function ( ) {
76+ let data = [ "dog" , "cat" , "man" , "dog" , "cat" , "man" , "man" , "cat" ] ;
77+ let series = new Series ( data ) ;
78+ let encode = new OneHotEncoder ( ) ;
79+ let fit_data = [
80+ [ 1 , 0 , 0 ] ,
81+ [ 0 , 1 , 0 ] ,
82+ [ 0 , 0 , 1 ] ,
83+ [ 1 , 0 , 0 ] ,
84+ [ 0 , 1 , 0 ] ,
85+ [ 0 , 0 , 1 ] ,
86+ [ 0 , 0 , 1 ] ,
87+ [ 0 , 1 , 0 ]
88+ ] ;
89+ let transform_data = [ [ 0 , 0 , 1 ] , [ 0 , 1 , 0 ] ] ;
9090
91- assert . deepEqual ( encode . fit ( series ) . values , fit_data ) ;
92- assert . deepEqual ( encode . transform ( new Series ( [ "man" , "cat" ] ) ) . values , transform_data )
93- } )
91+ assert . deepEqual ( encode . fit ( series ) . values , fit_data ) ;
92+ assert . deepEqual ( encode . transform ( new Series ( [ "man" , "cat" ] ) ) . values , transform_data ) ;
9493 } ) ;
94+ } ) ;
9595
9696
97- } ) ;
97+ } ) ;
0 commit comments