From 40a9a539971ed3755c3e2f237c87a57425415e54 Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Wed, 3 Feb 2021 14:48:25 +0800 Subject: [PATCH] init --- mllib/src/main/scala/org/apache/spark/ml/feature/Selector.scala | 2 +- .../scala/org/apache/spark/mllib/feature/ChiSqSelector.scala | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/Selector.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/Selector.scala index cb8b71aa5ba04..e24593a01b629 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/feature/Selector.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/feature/Selector.scala @@ -247,7 +247,7 @@ private[ml] abstract class Selector[T <: SelectorModel[T]] .where(col("pValue") < $(fwe) / numFeatures) .as[Int].collect() case errorType => - throw new IllegalStateException(s"Unknown Selector Type: $errorType") + throw new IllegalArgumentException(s"Unknown Selector Type: $errorType") } copyValues(createSelectorModel(uid, indices.sorted) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/feature/ChiSqSelector.scala b/mllib/src/main/scala/org/apache/spark/mllib/feature/ChiSqSelector.scala index 70125d2c4c6af..3202f08e220b0 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/feature/ChiSqSelector.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/feature/ChiSqSelector.scala @@ -283,7 +283,7 @@ class ChiSqSelector @Since("2.1.0") () extends Serializable { chiSqTestResult .filter { case (res, _) => res.pValue < fwe / chiSqTestResult.length } case errorType => - throw new IllegalStateException(s"Unknown ChiSqSelector Type: $errorType") + throw new IllegalArgumentException(s"Unknown ChiSqSelector Type: $errorType") } val indices = features.map { case (_, index) => index } new ChiSqSelectorModel(indices)