diff options
author | Yanbo Liang <ybliang8@gmail.com> | 2016-04-25 14:08:41 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2016-04-25 14:08:41 -0700 |
commit | 9cb3ba1013a7eae11be8a00fa4a9c5308bb20195 (patch) | |
tree | eb275db612f3bc4f438aa426bb49c528d6fc0fe9 /R/pkg/NAMESPACE | |
parent | 0c47e274ab8c286498fa002e2c92febcb53905c6 (diff) | |
download | spark-9cb3ba1013a7eae11be8a00fa4a9c5308bb20195.tar.gz spark-9cb3ba1013a7eae11be8a00fa4a9c5308bb20195.tar.bz2 spark-9cb3ba1013a7eae11be8a00fa4a9c5308bb20195.zip |
[SPARK-14312][ML][SPARKR] NaiveBayes model persistence in SparkR
## What changes were proposed in this pull request?
SparkR ```NaiveBayesModel``` supports ```save/load``` by the following API:
```
df <- createDataFrame(sqlContext, infert)
model <- naiveBayes(education ~ ., df, laplace = 0)
ml.save(model, path)
model2 <- ml.load(path)
```
## How was this patch tested?
Add unit tests.
cc mengxr
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #12573 from yanboliang/spark-14312.
Diffstat (limited to 'R/pkg/NAMESPACE')
-rw-r--r-- | R/pkg/NAMESPACE | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/R/pkg/NAMESPACE b/R/pkg/NAMESPACE index 0f92b5e597..c0a63d6b3e 100644 --- a/R/pkg/NAMESPACE +++ b/R/pkg/NAMESPACE @@ -107,7 +107,8 @@ exportMethods("arrange", "write.jdbc", "write.json", "write.parquet", - "write.text") + "write.text", + "ml.save") exportClasses("Column") @@ -299,7 +300,8 @@ export("as.DataFrame", "tableNames", "tables", "uncacheTable", - "print.summary.GeneralizedLinearRegressionModel") + "print.summary.GeneralizedLinearRegressionModel", + "ml.load") export("structField", "structField.jobj", |