Skip to content

Instantly share code, notes, and snippets.

@kaysush

kaysush/App.scala

Last active May 16, 2018 15:32
Show Gist options
  • Save kaysush/b4b4a9643cb90a6a2b434f6a0d0ddb13 to your computer and use it in GitHub Desktop.
Save kaysush/b4b4a9643cb90a6a2b434f6a0d0ddb13 to your computer and use it in GitHub Desktop.
Download ZIP
Assembling Pipeline
Raw
App.scala
// Assemble the pipeline
val stages = Array(genderIndexer, genderOneHotEncoder, vectorAssembler, scaler, logisticRegression)
val pipeline = new Pipeline().setStages(stages)
//Fit the pipeline
val model = pipeline.fit(train)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment