葡萄酒逻辑回归分类（scala实现）

最新推荐文章于 2022-06-06 10:58:44 发布

路易三十六

最新推荐文章于 2022-06-06 10:58:44 发布

阅读量2.6k

点赞数

分类专栏：机器学习文章标签：葡萄酒的评价数模葡萄酒的评价代码 scala 逻辑回归

本文链接：https://blog.csdn.net/LuYi_WeiLin/article/details/84676117

版权

本文介绍了使用Scala进行葡萄酒质量分类的方法，通过逻辑回归算法，将评分7.0以上定义为好葡萄酒，其他为坏葡萄酒。

摘要由CSDN通过智能技术生成

葡萄酒分类（scala实现）

分类方法：逻辑回归

其中

0代表坏葡萄酒
1代表好葡萄酒

训练集中质量评分7.0以上被视为好葡萄酒


	import org.apache.spark.ml.classification.LogisticRegression
	import org.apache.spark.ml.param.ParamMap
	import org.apache.spark.ml.linalg.{Vector, Vectors}
	import org.apache.spark.ml.regression.LinearRegressionModel
	import org.apache.spark.sql.{Row, SparkSession}

	object LogicRegressWineClassifyDemo {
		def main(args: Array[String]): Unit = {
			val sess = SparkSession.builder().appName("ml").master("local[4]").getOrCreate();
			val sc = sess.sparkContext;
			//数据目录
			val dataDir = "file:///D:/downloads/bigdata/ml/winequality-white.csv"
			//定义样例类
			case class Wine(FixedAcidity: Double, VolatileAcidity: Double,
							CitricAcid: Double, ResidualSugar: Double, Chlorides: Double,
							FreeSulfurDioxide: Double, Tota