Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- val trainingDataTable = sql("""
- SELECT e.action, u.age, u.latitude, u.longitude
- FROM Users u
- JOIN Events e
- ON u.userId = e.userId""")
- // Since `sql` returns an RDD, the results can be easily used in MLlib
- val trainingData = trainingDataTable.map { case Row(_, age, latitude, longitude) =>
- val features = Array[Double](age, latitude, longitude)
- LabeledPoint(age, features)
- }
- val model = new LogisticRegressionWithSGD().run(trainingData)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement