Apache Spark MLLib获得最大价值

Question

我有以下模型：

case class Product(price:Int,distance:Int)

并且我有数据告诉我，如果距离为y（是/否），那么客户是否愿意以价格x购买产品。

我对它进行了逻辑回归，现在可以预测（价格，距离）对。 如果我现在想知道距离x可以收取的最高价格怎么办？

码：

val products:List[(Product,Double)] = getProductVotes()
val points:List[LabeledPoints] = products.map{ case (product,vote) => 
        LabeledPoint(vote,Vectors.dense(product.price,product.distance)) }
val data: RDD[LabeledPoint] = sc.parallelize(points)
val splits = data.randomSplit(Array(0.6, 0.4), seed = 11L)
val training = splits(0).cache()
val test = splits(1).cache()
val model = new LogisticRegressionWithLBFGS()
    .setNumClasses(10)
    .run(training)

Answer 1

知道给定距离X的最高价格。

1. 取您的训练数据的子集，如果投票=真
2. 构建标签点，标签为“价格”，特征为“距离”
3. 在标记点的集合上训练线性回归模型，以在给定“距离”的情况下预测“价格”