Skip to content

Mgla96/AuthorPredictor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Author Predictor

This is a supervised learning author predictor program which I built to gain more knowledge in machine learning. This program utilizes Multinomial Naive Bayes Multi Classification with a bag-of-words representation of the text. I implemented bigrams and trigrams as well as used the nltk library to incorporate lemmetization as methods to further improve my accuracy. The accuracy could be further improved by implementing other features such as article length, using stop words, adding tf-idf weighting, and optimizing current weighting of unigram,bigram,and trigrams. My program is created with an empty stop words array which could be filled with words that we would like to ignore. Even more accuracy improvements could potentially be seen by using a word to vector approach which would allow us to implement different machine learning models altogether such as a Support Vector Machines or Neural Networks.

About

Predicts Anonymous Authors with Multinomial Naive Bayes Multi Classification

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages