Skip to content

Instantly share code, notes, and snippets.

@kaysush

kaysush/App.scala

Created May 16, 2018 15:25
Show Gist options
  • Save kaysush/b52d33752b2bfc19bb01e3bac9fde95b to your computer and use it in GitHub Desktop.
Save kaysush/b52d33752b2bfc19bb01e3bac9fde95b to your computer and use it in GitHub Desktop.
Download ZIP
Vector Assembler and Feature Scaling
Raw
App.scala
val features = Array("GenderOHE", "Age", "EstimatedSalary")
val dependetVariable = "Purchased"
val vectorAssembler = new VectorAssembler().setInputCols(features).setOutputCol("features")
val scaler = new StandardScaler().setInputCol("features").setOutputCol("scaledFeatures")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment