Skip to content

Commit fa029bb

Browse files
author
DB Tsai
committed
made the bound smaller
1 parent 0806002 commit fa029bb

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -423,10 +423,10 @@ class LogisticRegressionSuite extends FunSuite with MLlibTestSparkContext {
423423
val interceptR = 0.57734851
424424
val weightsR = Array(-0.05310287, 0.0, -0.08849250, -0.15458796)
425425

426-
assert(model.intercept ~== interceptR relTol 1E-2)
427-
assert(model.weights(0) ~== weightsR(0) relTol 1E-2)
426+
assert(model.intercept ~== interceptR relTol 6E-3)
427+
assert(model.weights(0) ~== weightsR(0) relTol 5E-3)
428428
assert(model.weights(1) ~== weightsR(1) relTol 1E-3)
429-
assert(model.weights(2) ~== weightsR(2) relTol 1E-2)
429+
assert(model.weights(2) ~== weightsR(2) relTol 5E-3)
430430
assert(model.weights(3) ~== weightsR(3) relTol 1E-3)
431431
}
432432

@@ -492,7 +492,7 @@ class LogisticRegressionSuite extends FunSuite with MLlibTestSparkContext {
492492
val interceptTheory = Math.log(histogram(1).toDouble / histogram(0).toDouble)
493493
val weightsTheory = Array(0.0, 0.0, 0.0, 0.0)
494494

495-
assert(model.intercept ~== interceptTheory relTol 1E-3)
495+
assert(model.intercept ~== interceptTheory relTol 1E-5)
496496
assert(model.weights(0) ~== weightsTheory(0) absTol 1E-6)
497497
assert(model.weights(1) ~== weightsTheory(1) absTol 1E-6)
498498
assert(model.weights(2) ~== weightsTheory(2) absTol 1E-6)
@@ -518,7 +518,7 @@ class LogisticRegressionSuite extends FunSuite with MLlibTestSparkContext {
518518
val interceptR = -0.248065
519519
val weightsR = Array(0.0, 0.0, 0.0, 0.0)
520520

521-
assert(model.intercept ~== interceptR relTol 1E-3)
521+
assert(model.intercept ~== interceptR relTol 1E-5)
522522
assert(model.weights(0) ~== weightsR(0) absTol 1E-6)
523523
assert(model.weights(1) ~== weightsR(1) absTol 1E-6)
524524
assert(model.weights(2) ~== weightsR(2) absTol 1E-6)

0 commit comments

Comments
 (0)