Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
zjd
 
 
 
 
 
 
 
 
 
 

CS5228 Final Project - News Article Classification

Project webpage: https://www.kaggle.com/c/cs5228-article-category-prediction

Project instructor:

Replicate out experiment

  • python model-1.py

    Use this to train model 1 and make prediction. (with only title)

  • python model-2.py

    Use this to train model 2 and make prediction. (with title and webpage content)

  • python ensemble-rule-based.py

    Our rule based ensemble method

  • python check-replication.py

    You can use this to check if you(also we) can replicate our best work on Kaggle, the answer is YES!

Dataset to contribute

We contribute our obtained webpage content, you can use this to push your model's performance to a higher level! Our team is really excited about that!

Usage: It is a pickle file, after load from pickle, it is a list, the index is in the same order as that in the original dataset.

Link:

https://github.com/YisongMiao/CS5228-project/blob/master/data/train_v2.pk

https://github.com/YisongMiao/CS5228-project/blob/master/data/test_v2.pk

Study purpose

Give us a star on this git repo if you like it!

Contact:

About

Winning 2nd place🥈at NUS CS5228 in-class Kaggle competition 2018!

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published