MLUtils.loadLibSVMFile

1 import org.apache.spark.mllib.util.MLUtils// Load and parse the data file.
2 
3 val data =
4 
5 MLUtils.loadLibSVMFile(sc, "data/mllib/sample_libsvm_data.txt")
6 
7 // Split data into training/test sets

 

转载于:https://www.cnblogs.com/suanec/p/4786347.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
假设你的数据集是以data为后缀的,比如traindata.csv和testdata.csv,其中traindata.csv包含y,x1,x2,x3的数据,testdata.csv只包含x1,x2,x3的数据。下面是使用pyspark mllib进行拆分和线性回归训练的代码: ```python from pyspark import SparkConf, SparkContext from pyspark.mllib.regression import LabeledPoint, LinearRegressionWithSGD from pyspark.mllib.util import MLUtils # 初始化Spark配置和SparkContext conf = SparkConf().setAppName("LinearRegression") sc = SparkContext(conf=conf) # 读取训练集和测试集 trainData = MLUtils.loadLibSVMFile(sc, "traindata.csv") testData = MLUtils.loadLibSVMFile(sc, "testdata.csv") # 把训练集和测试集分别拆分成y和x1,x2,x3 trainLabels = trainData.map(lambda lp: lp.label) trainFeatures = trainData.map(lambda lp: lp.features) testFeatures = testData.map(lambda lp: lp.features) # 把训练集分成训练集和验证集 (trainingData, validationData) = trainData.randomSplit([0.7, 0.3]) # 训练线性回归模型 model = LinearRegressionWithSGD.train(trainingData, iterations=100, step=0.1) # 使用验证集评估模型 valuesAndPreds = validationData.map(lambda p: (p.label, model.predict(p.features))) MSE = valuesAndPreds.map(lambda vp: (vp[0] - vp[1]) ** 2).reduce(lambda x, y: x + y) / valuesAndPreds.count() print("Mean Squared Error = " + str(MSE)) # 在测试集上预测 predictions = model.predict(testFeatures).collect() # 输出预测结果 print("Predictions: " + str(predictions)) # 停止SparkContext sc.stop() ``` 这段代码首先读取了训练集和测试集,然后把训练集和测试集分别拆分成y和x1,x2,x3。接着把训练集分成训练集和验证集,使用LinearRegressionWithSGD.train训练线性回归模型,使用验证集评估模型的性能。最后在测试集上预测,输出预测结果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值