Load & merge Dataset

→ train - bodies.csv

→ train - stones.csv

Level mapping

create combined Text

Heading + [SEP] + article Body

Train - Test split

Train - 80%

Test - 20%

model-specific Processing

Bert

• Tokenization

• TF Bert model

svm

• TF-IDF

• svm-classifier

RF

• TF-IDF

• Random Forest classifier

DF

• TF-IDF

• Decision Tree classifier

Model Training

Evaluation & output