@@ -143,7 +143,7 @@ public void testPreprocessCols() throws Exception {
143
143
// TODO: correct `binEdges` of feature `f0` after FLINK-30734 resolved.
144
144
List <FeatureMeta > expectedMeta =
145
145
Arrays .asList (
146
- FeatureMeta .continuous ("f0" , 3 , new double [] {1.2 , 4.5 , 13.9 , Double . NaN }),
146
+ FeatureMeta .continuous ("f0" , 3 , new double [] {1.2 , 4.5 , 13.9 , 15.3 }),
147
147
FeatureMeta .continuous ("f1" , 3 , new double [] {1.0 , 2.0 , 4.0 , 5.0 }),
148
148
FeatureMeta .categorical ("f2" , 5 , new String [] {"a" , "b" , "c" , "d" , "e" }));
149
149
@@ -154,7 +154,7 @@ public void testPreprocessCols() throws Exception {
154
154
Row .of (40.0 , 0 , 2 , 2.0 ),
155
155
Row .of (40.0 , 1 , 2 , 0.0 ),
156
156
Row .of (40.0 , 1 , 1 , 1.0 ),
157
- Row .of (41.0 , 2 , 1 , 2.0 ),
157
+ Row .of (41.0 , 3 , 1 , 2.0 ),
158
158
Row .of (41.0 , 1 , 2 , 4.0 ),
159
159
Row .of (41.0 , 2 , 1 , 1.0 ),
160
160
Row .of (41.0 , 2 , 2 , 0.0 ),
@@ -188,19 +188,17 @@ public void testPreprocessVectorCol() throws Exception {
188
188
// TODO: correct `binEdges` of feature `_vec_f0` and `_vec_f2` after FLINK-30734 resolved.
189
189
List <FeatureMeta > expectedMeta =
190
190
Arrays .asList (
191
- FeatureMeta .continuous (
192
- "_vec_f0" , 3 , new double [] {1.2 , 4.5 , 13.9 , Double .NaN }),
191
+ FeatureMeta .continuous ("_vec_f0" , 3 , new double [] {1.2 , 4.5 , 13.9 , 15.3 }),
193
192
FeatureMeta .continuous ("_vec_f1" , 3 , new double [] {1.0 , 2.0 , 4.0 , 5.0 }),
194
- FeatureMeta .continuous (
195
- "_vec_f2" , 3 , new double [] {1.0 , 2.0 , 3.0 , Double .NaN }));
193
+ FeatureMeta .continuous ("_vec_f2" , 3 , new double [] {1.0 , 2.0 , 3.0 , 5.0 }));
196
194
List <Row > expectedPreprocessedRows =
197
195
Arrays .asList (
198
- Row .of (40.0 , Vectors .dense (0 , 1 , 2 .0 )),
196
+ Row .of (40.0 , Vectors .dense (0 , 1 , 3 .0 )),
199
197
Row .of (40.0 , Vectors .dense (0 , 1 , 1.0 )),
200
198
Row .of (40.0 , Vectors .dense (0 , 2 , 2.0 )),
201
199
Row .of (40.0 , Vectors .dense (1 , 2 , 0.0 )),
202
200
Row .of (40.0 , Vectors .dense (1 , 1 , 1.0 )),
203
- Row .of (41.0 , Vectors .dense (2 , 1 , 2.0 )),
201
+ Row .of (41.0 , Vectors .dense (3 , 1 , 2.0 )),
204
202
Row .of (41.0 , Vectors .dense (1 , 2 , 2.0 )),
205
203
Row .of (41.0 , Vectors .dense (2 , 1 , 1.0 )),
206
204
Row .of (41.0 , Vectors .dense (2 , 2 , 0.0 )),
0 commit comments