You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
At the moment the models trained for science ensemble are trained independently. Since there is a model class ScienceEnsemble the training should move inside fit which will simplify dvc.yaml as well. Note that save will also need to be implemented.
There are two ways for implementing that
through the use of WellcomeVotingClassifier which would be ideal
through training the classifier independently inside fit and saving them
The WellcomeVotingClassifier might not be possible due to some internals of sklearn but we should investigate whether it makes sense to adjust the code to ensure it works. tfidf-svm is already an sklearn pipeline so any problems will probably arise from BertClassifier
The text was updated successfully, but these errors were encountered:
At the moment the models trained for science ensemble are trained independently. Since there is a model class
ScienceEnsemble
the training should move insidefit
which will simplifydvc.yaml
as well. Note thatsave
will also need to be implemented.There are two ways for implementing that
WellcomeVotingClassifier
which would be idealThe
WellcomeVotingClassifier
might not be possible due to some internals of sklearn but we should investigate whether it makes sense to adjust the code to ensure it works.tfidf-svm
is already an sklearn pipeline so any problems will probably arise fromBertClassifier
The text was updated successfully, but these errors were encountered: