支持向量机

来源:互联网 发布:战天堂翅膀进阶数据 编辑:程序博客网 时间:2024/06/06 14:24

一、支持向量机使用示例

数据:

1|21|31|41|51|60|70|80|90|100|11
程序:

import org.apache.spark.mllib.linalg.Vectorsimport org.apache.spark.{SparkConf, SparkContext}import org.apache.spark.mllib.classification.SVMWithSGD;import org.apache.spark.mllib.regression.LabeledPoint;object kimiYang {  var conf = new SparkConf()    .setMaster("local")    .setAppName("kimiYang");  var sc = new SparkContext(conf);  def main(args: Array[String]): Unit = {    val data = sc.textFile("kimi.txt") //读取数据文件    val parsedData = data.map{line =>      val parts = line.split('|');      LabeledPoint(parts(0).toDouble, Vectors.dense(parts(1).split(' ')map(_.toDouble)))    }.cache();    val model = SVMWithSGD.train(parsedData,10);    println(model.weights);权重    println(model.intercept);截距  }}
结果:

[-0.9034130025957295]
0.0


0 0
原创粉丝点击