Merge pull request #972 from warrenzhu25/patch-1

Fix sparkml example
This commit is contained in:
Umachandar Jayachandran
2022-03-02 22:07:34 -08:00
committed by GitHub
@@ -340,7 +340,7 @@
" featureCols.append(key)\r\n",
"\r\n",
"# string-index the label column into a column named \"label\"\r\n",
"si_label = StringIndexer(inputCol=label, outputCol='label')\r\n",
"si_label = StringIndexer(inputCol=label, outputCol='label').setHandleInvalid("skip")\r\n",
"\r\n",
"# assemble the encoded feature columns in to a column named \"features\"\r\n",
"assembler = VectorAssembler(inputCols=featureCols, outputCol=\"features\")\r\n",