Skip to content

Siddhant128-bit/Classify_PDF_Multinominal_Naive_Bayes

Repository files navigation

Document Classification System using Multinomial Naive Bayes Theorem


This note is to log the development process of the overall system. So let's begin

What is this system about ?

   The idea is we are going to build a system which can read any pdf and build a profile using vectorizer like count vectorizer this will make our system capable of understanding what kind of words are there in such document type and hence will be able to predict any new document as one of the previous datatypes


How does it work

   Initially we train our model to understand datas from past which are labeled as particular topic realted or not. Then we are going to use various unknown data for inference testing and see how well it performs.


Where can it work

   Our system as of now works only with pdf documents but we can make it adaptable to work with any datatype provided we can read the text from that particular file.


Let's learn more


Multinomial Naive Bayes

We are going to take a bunch of words convert it into dataset having words and title such that the first column of dataste consists all the words required for the particular task of classification and title consists all the label assigned to a particular class.


    Libraries used
  • Pymupdf (fitz)
  • Pandas
  • Scikit-learn


  • First we are going to use Preprocess to create folders where all files are stored then we are going to use Dataset Generator to generate an excell file with all the details that we are going to use for training purpose. Finally we are going to use classifier.py for classification purpose of the files



    This bag of words approach can be swtiched with tfidf approach to test for new possible words and working on how it can be used to make a better classifcation model for special terms and words



    This project can be scaled up to include deep learning approach to have feature extraction in and of itself.


    Documentation on Progress !!

    About

    This model is basically to be able to classify pdfs based on the contents currently its rigid in structure can be made more flexible.

    Resources

    Stars

    3 stars

    Watchers

    1 watching

    Forks

    Releases

    No releases published

    Packages

     
     
     

    Contributors

    Languages