An open API service indexing awesome lists of open source software.

https://github.com/mgla96/authorpredictor

Predicts Anonymous Authors with Multinomial Naive Bayes Multi Classification
https://github.com/mgla96/authorpredictor

Last synced: 3 months ago
JSON representation

Predicts Anonymous Authors with Multinomial Naive Bayes Multi Classification

Awesome Lists containing this project

README

          

# 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.