diff options
Diffstat (limited to 'mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala index da51cb7800..c41fb4b062 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala @@ -57,7 +57,7 @@ import org.apache.spark.sql.functions._ */ @Since("1.4.0") @Experimental -final class GBTRegressor @Since("1.4.0") (@Since("1.4.0") override val uid: String) +class GBTRegressor @Since("1.4.0") (@Since("1.4.0") override val uid: String) extends Predictor[Vector, GBTRegressor, GBTRegressionModel] with GBTRegressorParams with DefaultParamsWritable with Logging { @@ -157,7 +157,7 @@ object GBTRegressor extends DefaultParamsReadable[GBTRegressor] { */ @Since("1.4.0") @Experimental -final class GBTRegressionModel private[ml]( +class GBTRegressionModel private[ml]( override val uid: String, private val _trees: Array[DecisionTreeRegressionModel], private val _treeWeights: Array[Double], |