Elephas models for Spark ML

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.

elephas.ml_model.ElephasTransformer()

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

elephas.ml_model.load_ml_transformer(file_name)

elephas.ml_model.load_ml_estimator(file_name)