Skip to content

lassesofus/MLOPS-Exam-Project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fake news detection

==============================

Description of the project

Following the intention of the Kaggle competition, we aim to build a system capable of classifying unreliable news articles.

Framework

We plan on using the Huggingface Transformer framework. We will be using a pre-trained model included in the framework.

Data description

The data is made available through the competition. This dataset contains the following features for each news article: A unique ID, the title, the author, the textual content, and finally a binary label of 0 or 1 corresponding to “reliable” and unreliable respectively. The model will be trained using DTU HPC.

Models

We expect to use a transformer model made for Natural Language Processing - specifically a pre-trained BERT. The model used is the base version with 12 layers and 12 self-attention heads. The hidden size is 768.

Usage

To use this project, you will need to install the required packages listed in requirements.txt. You can then run the scripts in the src directory to process the data, build features, train models, and make predictions. The Makefile includes commands for running these scripts, such as make data, make train, and make evaluate.

License

This project is licensed under the terms of the MIT License.

Project based on the cookiecutter data science project template. #cookiecutterdatascience

About

Repository for the exam project for the DTU MLOps course

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 6