Elephas models for Spark ML

elephas.ml_model.ElephasEstimator(**kwargs)

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(**kwargs)

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: str)

Load a Transformer from a HDF5 file. :param file_name: :return:

elephas.ml_model.load_ml_estimator(file_name: str)