diff options
Diffstat (limited to 'mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala index 8eaed8b682..b6ab2fd625 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala @@ -43,7 +43,7 @@ import org.apache.spark.sql.functions._ */ @Since("1.4.0") @Experimental -final class RandomForestRegressor @Since("1.4.0") (@Since("1.4.0") override val uid: String) +class RandomForestRegressor @Since("1.4.0") (@Since("1.4.0") override val uid: String) extends Predictor[Vector, RandomForestRegressor, RandomForestRegressionModel] with RandomForestRegressorParams with DefaultParamsWritable { @@ -137,7 +137,7 @@ object RandomForestRegressor extends DefaultParamsReadable[RandomForestRegressor */ @Since("1.4.0") @Experimental -final class RandomForestRegressionModel private[ml] ( +class RandomForestRegressionModel private[ml] ( override val uid: String, private val _trees: Array[DecisionTreeRegressionModel], override val numFeatures: Int) |