diff options
Diffstat (limited to 'examples/src/main/python/ml/rformula_example.py')
-rw-r--r-- | examples/src/main/python/ml/rformula_example.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/examples/src/main/python/ml/rformula_example.py b/examples/src/main/python/ml/rformula_example.py index d5df3ce4f5..6629239db2 100644 --- a/examples/src/main/python/ml/rformula_example.py +++ b/examples/src/main/python/ml/rformula_example.py @@ -34,10 +34,12 @@ if __name__ == "__main__": (8, "CA", 12, 0.0), (9, "NZ", 15, 0.0)], ["id", "country", "hour", "clicked"]) + formula = RFormula( formula="clicked ~ country + hour", featuresCol="features", labelCol="label") + output = formula.fit(dataset).transform(dataset) output.select("features", "label").show() # $example off$ |