2

Ich möchte Koeffizientenwerte des linearen Regressionsmodells (LR) in Spark-MLlib erhalten. Hier verwende ich den ‚LinearRegressionWithSGD‘ um das Modell zu bauen und man kann die Probe aus dem folgenden Link:Wie erhalten Sie Koeffizientenwerte vom linearen Spark-MLlib-Regressionsmodell (Scala)?

https://spark.apache.org/docs/2.1.0/mllib-linear-methods.html#regression

Ich kann die Koeffizientenwerte Spark-ML linearer Regression erhalten. Bitte finden Sie den Referenzlink von unten.

https://spark.apache.org/docs/2.1.0/ml-classification-regression.html#linear-regression

Bitte helfen Sie mir mit diesem. Danke im Voraus !!

Antwort

1

ersten Zeilen Modellerstellung aus dem ersten Link gesendet Nahm:

val model: LinearRegressionModel = LinearRegressionWithSGD.train(parsedData, numIterations, stepSize) 
.run(training) 
// Here are the coefficient and intercept 
val weights: org.apache.spark.mllib.linalg.Vector = model.weights 
val intercept = model.intercept 
val weightsData: Array[Double] = weights.asInstanceOf[DenseVector].values 

Die letzten drei Zeilen der Koeffizient und Intercept

Die Art der weights sind, ist : org.apache.spark.mllib.linalg.Vector

Das heißt ein Wrapper um den Breeze DenseVector

+0

Gibt es eine Möglichkeit, den Spalten-/Feldnamen von th zu finden e Koeffizient? Hier geben die "Gewichte" die Koeffizienten als Array zurück und ich bin mir nicht sicher, welcher Spaltenkoeffizient es ist? Kannst du mir bitte helfen? – Ramkumar

+0

Ich möchte einfach wissen, welche Felder ich für die Erstellung eines linearen Regressionsmodells berücksichtigt habe. Gibt es eine API? – Ramkumar

+0

Ich habe das eine Weile nicht angeschaut. Ich denke, ich habe gerade ein einfaches Testprojekt in Scala erstellt und die Objekte aus Eclipse betrachtet. –

Verwandte Themen