The objective of this challenge is to develop a machine learning model that classifies statements and questions expressed by university students in Kenya when speaking about the mental health challenges they struggle with. The four categories are depression, suicide, alchoholism, and drug abuse. For more information about this challenge, have a look on Zindi.
|----bnbr (package)
| |--- ...
| |--- {module}.py
| |--- ...
|
|----data (placeholder for raw and preprocessed data)
| |--- Train.csv
| |--- Test.csv
| |--- SampleSubmission.csv
| |--- ...
|
|----mask_language_modeling
| |--- MLM_BertBase_finetuning.ipynb
| |--- MLM_RobertaBase_finetuning.ipynb
|
|----notebooks
| |--- BNBR_PreProcessing.ipynb.ipynb
| |--- TranslateWithTransformers.ipynb
| |--- MLMRobertaBaseGenericModel.ipynb
| |--- MLMBertBaseGenericModel.ipynb
| |--- Blend_roberta_roberta_bert.ipynb
|
|----translated
| |--- TRANSLATE_THE_DATA.ipynb
| |--- second_roberta.ipynb
|
|---- Readme.md