Advertisement
Guest User

Untitled

a guest
Jul 22nd, 2019
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 4.40 KB | None | 0 0
  1. // StringIndexer define new 'label' column with 'result' column
  2. val indexer = new StringIndexer()
  3. .setInputCol("creditability")
  4. .setOutputCol("label")
  5. val labelDf = indexer.fit(featureDf).transform(featureDf)
  6. labelDf.printSchema()
  7. /*
  8. * outout
  9. root
  10. |-- creditability: double (nullable = true)
  11. |-- balance: double (nullable = true)
  12. |-- duration: integer (nullable = true)
  13. |-- history: integer (nullable = true)
  14. |-- purpose: integer (nullable = true)
  15. |-- amount: integer (nullable = true)
  16. |-- savings: integer (nullable = true)
  17. |-- employment: integer (nullable = true)
  18. |-- instPercent: integer (nullable = true)
  19. |-- sexMarried: integer (nullable = true)
  20. |-- guarantors: integer (nullable = true)
  21. |-- residenceDuration: integer (nullable = true)
  22. |-- assets: integer (nullable = true)
  23. |-- age: integer (nullable = true)
  24. |-- concCredit: integer (nullable = true)
  25. |-- apartment: integer (nullable = true)
  26. |-- credits: integer (nullable = true)
  27. |-- occupation: integer (nullable = true)
  28. |-- dependents: integer (nullable = true)
  29. |-- hasPhone: integer (nullable = true)
  30. |-- foreign: integer (nullable = true)
  31. |-- features: vector (nullable = true)
  32. |-- label: double (nullable = false)
  33. */
  34.  
  35. labelDf.show(10)
  36. /*
  37. * outout
  38. +-------------+-------+--------+-------+-------+------+-------+----------+-----------+----------+----------+-----------------+------+---+----------+---------+-------+----------+----------+--------+-------+--------------------+-----+
  39. |creditability|balance|duration|history|purpose|amount|savings|employment|instPercent|sexMarried|guarantors|residenceDuration|assets|age|concCredit|apartment|credits|occupation|dependents|hasPhone|foreign| features|label|
  40. +-------------+-------+--------+-------+-------+------+-------+----------+-----------+----------+----------+-----------------+------+---+----------+---------+-------+----------+----------+--------+-------+--------------------+-----+
  41. | 1.0| 1.0| 9| 4| 0| 2799| 1| 3| 2| 3| 1| 2| 1| 36| 3| 1| 2| 3| 2| 1| 1|[1.0,9.0,4.0,0.0,...| 0.0|
  42. | 1.0| 2.0| 12| 2| 9| 841| 2| 4| 2| 2| 1| 4| 1| 23| 3| 1| 1| 2| 1| 1| 1|[2.0,12.0,2.0,9.0...| 0.0|
  43. | 1.0| 1.0| 12| 4| 0| 2122| 1| 3| 3| 3| 1| 2| 1| 39| 3| 1| 2| 2| 2| 1| 2|[1.0,12.0,4.0,0.0...| 0.0|
  44. | 1.0| 1.0| 12| 4| 0| 2171| 1| 3| 4| 3| 1| 4| 2| 38| 1| 2| 2| 2| 1| 1| 2|[1.0,12.0,4.0,0.0...| 0.0|
  45. | 1.0| 1.0| 10| 4| 0| 2241| 1| 2| 1| 3| 1| 3| 1| 48| 3| 1| 2| 2| 2| 1| 2|[1.0,10.0,4.0,0.0...| 0.0|
  46. | 1.0| 1.0| 8| 4| 0| 3398| 1| 4| 1| 3| 1| 4| 1| 39| 3| 2| 2| 2| 1| 1| 2|[1.0,8.0,4.0,0.0,...| 0.0|
  47. | 1.0| 1.0| 6| 4| 0| 1361| 1| 2| 2| 3| 1| 4| 1| 40| 3| 2| 1| 2| 2| 1| 2|[1.0,6.0,4.0,0.0,...| 0.0|
  48. | 1.0| 4.0| 18| 4| 3| 1098| 1| 1| 4| 2| 1| 4| 3| 65| 3| 2| 2| 1| 1| 1| 1|[4.0,18.0,4.0,3.0...| 0.0|
  49. | 1.0| 2.0| 24| 2| 3| 3758| 3| 1| 1| 2| 1| 4| 4| 23| 3| 1| 1| 1| 1| 1| 1|[2.0,24.0,2.0,3.0...| 0.0|
  50. | 1.0| 1.0| 11| 4| 0| 3905| 1| 3| 2| 3| 1| 2| 1| 36| 3| 1| 2| 3| 2| 1| 1|[1.0,11.0,4.0,0.0...| 0.0|
  51. +-------------+-------+--------+-------+-------+------+-------+----------+-----------+----------+----------+-----------------+------+---+----------+---------+-------+----------+----------+--------+-------+--------------------+-----+
  52. */
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement