FKarl / short-text-classification

This repository contains code to reproduce the results in our paper "Transformers are Short Text Classifiers: A Study of Inductive Short Text Classifiers on Benchmarks and Real-world Datasets".
MIT License
36 stars 7 forks source link

No module named 'ensemble_models' #3

Open magnolini opened 2 months ago

magnolini commented 2 months ago

I'm trying to do your tutorial but I'm having this error: Traceback (most recent call last): File ".../short-text-classification/main.py", line 17, in from ensemble_models import SimplifiedStacking, SimplifiedWeightedBoost, WeightedEnsemble ModuleNotFoundError: No module named 'ensemble_models'

I can't find this module anywhere, how can I solve this problem?

FKarl commented 2 months ago

Thank you for your message. The classes which are loaded here are in the Ensemble_models.py. This is most likely a refactor error as the file is capitalized but the import is lowercase. I can't fix it directly at the moment but probably only the import has to be capitalized. from Ensemble_models import ...