Elephas models for Spark ML
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.
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)