anthonymorast / pyml-ensemble

A Python package to handle ensembles of machine learning models.
MIT License
0 stars 0 forks source link

Accept numpy arrays (not lists) to ensemble.train #5

Open anthonymorast opened 3 years ago

anthonymorast commented 3 years ago

If all of the ensemble members should be trained on the same training data the training method should be able to take only the training x and y data rather than lists of the same training and test data. This would reduce confusion and memory footprint.

That is,

ensemble.train([trainx for _ in range(nummodels)], [trainy for in range(num_models)])

would become

ensemble.train(trainx, trainy)

anthonymorast commented 3 years ago

This would also prevent having to update the num_models variable in the train call when adding new models to the ensemble as all members would just use the same passed data.