scala随机划分及随机采样

随机划分及随机采样

  • 随机划分
  • 随机采样
  • 重新洗牌

随机划分

df.randomSplit(Array(0.8,0.2),1115)

df.randomSplit(Array(0.8,0.2),1115(0).count

df.randomSplit(Array(0.8,0.2),1115)(0).toDF.show//第一份转成DF,没有(0)不行

df.randomSplit(Array(0.01,0.9,0.09),1115)(0).toDF.count//可以分成多份,不一定是总和为1,最后的划分是array里面各个数的比例。比如Array(6,2,2)会分成0.6,0.2,0.2的三份

随机采样

val seed = 1115val withreplacement = false //是否有放回

val fraction = 0.2 //抽样比例

df.sample(withreplacement,fraction,seed).showdf.sample(withreplacement,fraction,seed).count

df.sample(false,0.001,seed).showdf.sample(false,0.001,seed).select("bat_batch").distinct.show

重新洗牌

df.shuffle
  • 7
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是使用 Scala 实现的随机森林算法示例代码,用于预测淘宝购物者是否会成为回头客: ```scala import org.apache.spark.SparkConf import org.apache.spark.SparkContext import org.apache.spark.mllib.tree.RandomForest import org.apache.spark.mllib.tree.model.RandomForestModel import org.apache.spark.mllib.util.MLUtils object RandomForestDemo { def main(args: Array[String]): Unit = { val conf = new SparkConf().setAppName("RandomForestDemo").setMaster("local") val sc = new SparkContext(conf) // 加载数据 val data = MLUtils.loadLibSVMFile(sc, "data/sample_libsvm_data.txt") // 将数据划分为训练集和测试集 val splits = data.randomSplit(Array(0.7, 0.3)) val (trainingData, testData) = (splits(0), splits(1)) // 训练模型 val numClasses = 2 val categoricalFeaturesInfo = Map[Int, Int]() val numTrees = 3 val featureSubsetStrategy = "auto" val impurity = "gini" val maxDepth = 5 val maxBins = 32 val model = RandomForest.trainClassifier(trainingData, numClasses, categoricalFeaturesInfo, numTrees, featureSubsetStrategy, impurity, maxDepth, maxBins) // 测试模型 val labelAndPreds = testData.map { point => val prediction = model.predict(point.features) (point.label, prediction) } val testErr = labelAndPreds.filter(r => r._1 != r._2).count.toDouble / testData.count() println("Test Error = " + testErr) println("Learned classification forest model:\n" + model.toDebugString) // 保存模型 model.save(sc, "myModelPath") // 加载模型 val sameModel = RandomForestModel.load(sc, "myModelPath") } } ``` 在这个示例中,我们使用了 Spark MLlib 库中的随机森林算法,通过加载样本数据、划分训练集和测试集、训练模型、测试模型以及保存和加载模型等步骤,实现了对淘宝购物者是否会成为回头客进行预测的功能。与决策树算法相比,随机森林算法能够进一步提升预测的准确性。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值