Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- amount gender_num marital_num
- 10000 1 1
- 20000 1 2
- 1400 2 1
- val df1 = df("loan_amount", 'loan_amount.cast("Double")).withColumn("gender_num", 'gender_num.cast("String")).
- withColumn("marital_num", 'marital_num.cast("String"))
- val labeled = df1.map(row => LabeledPoint(df1.gender_num, Vectors.dense(df1.loan_amount, df1.marital_num)))
- val numClasses = 7
- val categoricalFeaturesInfo = Map[Int, Int]()
- val numTrees = 3 // Use more in practice.
- val featureSubsetStrategy = "auto" // Let the algorithm choose.
- val impurity = "gini"
- val maxDepth = 4
- val maxBins = 32
- val model = RandomForest.trainClassifier(labeled, categoricalFeaturesInfo,
- numTrees, featureSubsetStrategy, impurity, maxDepth, maxBins)
- 138: error: value gender_num is not a member of org.apache.spark.sql.DataFrame
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement