@@ -80,29 +80,29 @@ class VectorsSuite extends FunSuite {
80
80
}
81
81
82
82
test(" sparse argmax" ) {
83
- val vec = Vectors .sparse(0 ,Array .empty[Int ],Array .empty[Double ]).asInstanceOf [SparseVector ]
83
+ val vec = Vectors .sparse(0 , Array .empty[Int ], Array .empty[Double ]).asInstanceOf [SparseVector ]
84
84
assert(vec.argmax === - 1 )
85
85
86
- val vec2 = Vectors .sparse(n,indices,values).asInstanceOf [SparseVector ]
86
+ val vec2 = Vectors .sparse(n, indices, values).asInstanceOf [SparseVector ]
87
87
assert(vec2.argmax === 3 )
88
88
89
- val vec3 = Vectors .sparse(5 ,Array (2 , 3 , 4 ),Array (1.0 , 0.0 , - .7 ))
89
+ val vec3 = Vectors .sparse(5 , Array (2 , 3 , 4 ), Array (1.0 , 0.0 , - .7 ))
90
90
assert(vec3.argmax === 2 )
91
91
92
92
// check for case that sparse vector is created with only negative values {0.0, 0.0,-1.0, -0.7, 0.0}
93
- val vec4 = Vectors .sparse(5 ,Array (2 , 3 ),Array (- 1.0 , - .7 ))
93
+ val vec4 = Vectors .sparse(5 , Array (2 , 3 ), Array (- 1.0 , - .7 ))
94
94
assert(vec4.argmax === 0 )
95
95
96
- val vec5 = Vectors .sparse(11 ,Array (0 , 3 , 10 ),Array (- 1.0 , - .7 , 0.0 ))
96
+ val vec5 = Vectors .sparse(11 , Array (0 , 3 , 10 ), Array (- 1.0 , - .7 , 0.0 ))
97
97
assert(vec5.argmax === 1 )
98
98
99
- val vec6 = Vectors .sparse(11 ,Array (0 , 1 , 2 ),Array (- 1.0 , - .7 , 0.0 ))
99
+ val vec6 = Vectors .sparse(11 , Array (0 , 1 , 2 ), Array (- 1.0 , - .7 , 0.0 ))
100
100
assert(vec6.argmax === 2 )
101
101
102
- val vec7 = Vectors .sparse(5 ,Array (0 , 1 , 3 ),Array (- 1.0 , 0.0 , - .7 ))
102
+ val vec7 = Vectors .sparse(5 , Array (0 , 1 , 3 ), Array (- 1.0 , 0.0 , - .7 ))
103
103
assert(vec7.argmax === 1 )
104
104
105
- var vec8 = Vectors .sparse(5 ,Array (1 , 2 ),Array (0.0 , - 1.0 ))
105
+ val vec8 = Vectors .sparse(5 , Array (1 , 2 ), Array (0.0 , - 1.0 ))
106
106
assert(vec8.argmax === 0 )
107
107
}
108
108
0 commit comments