支持向量机

一、支持向量机使用示例

数据：

1|2
1|3
1|4
1|5
1|6
0|7
0|8
0|9
0|10
0|11

程序：

import org.apache.spark.mllib.linalg.Vectors
import 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