Update the predict logic in the `SparkHFModel`

danielenricocahall / elephas

Distributed Deep learning with Keras & Spark

MIT License

17 stars 5 forks source link

Update the predict logic in the `SparkHFModel` #38

Closed danielenricocahall closed 2 months ago

danielenricocahall commented 2 months ago

Ensure if the user provides num_partitions, we partition and collect the results correctly. As this logic is identical to the logic in SparkModel, it has been consolidated into the _predict_and_collect method.