Skip to content
Snippets Groups Projects
Commit 21ad8462 authored by DB Tsai's avatar DB Tsai
Browse files

[MINOR][ML] removed the old `getModelWeights` function

Removed the old `getModelWeights` function which was private and renamed into `getModelCoefficients`

Author: DB Tsai <dbt@netflix.com>

Closes #9426 from dbtsai/feature-minor.
parent 476f4348
No related branches found
No related tags found
No related merge requests found
......@@ -51,16 +51,6 @@ private[r] object SparkRWrappers {
pipeline.fit(df)
}
@deprecated("Use getModelCoefficients instead.", "1.6.0")
def getModelWeights(model: PipelineModel): Array[Double] = {
model.stages.last match {
case m: LinearRegressionModel =>
Array(m.intercept) ++ m.weights.toArray
case m: LogisticRegressionModel =>
Array(m.intercept) ++ m.weights.toArray
}
}
def getModelCoefficients(model: PipelineModel): Array[Double] = {
model.stages.last match {
case m: LinearRegressionModel =>
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment