我试图在样本数据上构建Logistic回归模型。
我们可以得到的模型输出是用于构建模型的特征的权重。
我找不到Spark API用于估算的标准误差,Wald-Chi Square统计量,p值等。
我将下面的代码作为示例粘贴
import org.apache.spark.mllib.classification.LogisticRegressionWithLBFGS
import org.apache.spark.mllib.evaluation.{BinaryClassificationMetrics, MulticlassMetrics}
import org.apache.spark.mllib.linalg.Vectors
import org.apache.spark.mllib.regression.LabeledPoint
import org.apache.spark.mllib.tree.RandomForest
import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}
val sc = new SparkContext(new SparkConf().setAppName("SparkTest").setMaster("local[*]"))
val sqlContext = new org.apache.spark.sql.SQLContext(sc);
val data: RDD[String] = sc.textFile("C:/Users/user/Documents/spark-1.5.1-bin-hadoop2.4/data/mllib/credit_approval_2_attr.csv")
val parsedData = data.map { line =>
val parts = line.split(',').map(_.toDouble)
LabeledPoint(parts(0), Vectors.dense(parts.tail))
}
//Splitting the data
val splits: Array[RDD[LabeledPoint]] = parsedData.randomSplit(Array(0.7, 0.3), seed = 11L)
val training: RDD[LabeledPoint] = splits(0).cache()
val test: RDD[LabeledPoint] = splits(1)
// Run training algorithm to build the model
val model = new LogisticRegressionWithLBFGS()
.setNumClasses(2)
.run(training)
// Clear the prediction threshold so the model will return probabilities
model.clearThreshold
print(model.weights)
模型权重输出
[-0.03335987643613915,0.025215092730373874,0.22617842810253946,0.29415985532104943,-0.0025559467210279694,4.5242237280512646E-4]
只是一系列重量。
虽然我能够计算精度,召回率,准确度,灵敏度和其他模型诊断。
有没有办法可以计算出估计的标准误差,Wald-Chi Square统计量,Spark中的p值?
我很担心,因为R或SAS中有标准输出。
这是否必须对我们在Spark中使用的优化方法做些什么?
这里我们使用L-BFGS或SGD。
可能我不了解评估方法。
任何建议都将受到高度赞赏。
答案 0 :(得分:2)
以下方法将提供卡方检验的详细信息 -
<强> Statistics.chiSqTest(数据)强>
输入数据
val obs: RDD[LabeledPoint] =
sc.parallelize(
Seq(
LabeledPoint(1.0, Vectors.dense(1.0, 0.0, 3.0)),
LabeledPoint(1.0, Vectors.dense(1.0, 2.0, 0.0)),
LabeledPoint(-1.0, Vectors.dense(-1.0, 0.0, -0.5)
)
)
)
val featureTestResults: Array[ChiSqTestResult] = Statistics.chiSqTest(obs)
返回一个数组,其中包含针对标签的每个要素的ChiSquaredTestResult。
测试摘要,包括p值,自由度,检验统计量,使用的方法和零假设。