Elephas models for Spark ML

[source]

ElephasEstimator

elephas.ml_model.ElephasEstimator()

SparkML Estimator implementation of an elephas model. This estimator takes all relevant arguments for model compilation and training.

Returns a trained model in form of a SparkML Model, which is also a Transformer.


[source]

ElephasTransformer

elephas.ml_model.ElephasTransformer()

SparkML Transformer implementation. Contains a trained model, with which new feature data can be transformed into labels.


load_ml_transformer

elephas.ml_model.load_ml_transformer(file_name)

load_ml_estimator

elephas.ml_model.load_ml_estimator(file_name)