src.interface package

Submodules

src.interface.emailclassifierapp module

class src.interface.emailclassifierapp.EmailClassifierApp

Bases: object

Class that is the interface for EmailClassifier

classifier_option_check(option, constant)

Method that checks if provided classifier option is valid.

classify_emails_3_stage_pipelines(emails)

Classify one or multiple emails through 3-stage pipelines.

load_data_csv(path)

Method loads data from csv file. If it fails, an exception is raised.

predict_with_multiclassifier(emails)

Method predicts emails’ type with multiclassifier.

set_model1_clf(clf)

Method sets model object with provided classifier and returns it.

set_model2_clf(clf)

Method sets model object with provided classifier and returns it.

set_model3_clf(clf)

Method sets model object with provided classifier and returns it.

set_multiclassifier_model_clf(clf)

Method sets model object with provided classifier and returns it.

set_stacking_classifier_estimators(estimator_1, estimator_2, estimator_3)

Method allows user to set estimators for StackingClassifier.

set_voting_classifier_parameters(estimator_1, estimator_2, estimator_3, voting_option)

Method allows user to set estimators and voting option for VotingClassifier.

train_3_stage_pipelines(path1='../data/training_emails_stage_1.csv', path2='../data/training_emails_stage_2.csv', path3='../data/training_emails_stage_3.csv', classifier_option_1='MultinomialNB', classifier_option_2='MultinomialNB', classifier_option_3='MultinomialNB', column_name_1='related_to_jobhunt', column_name_2='is_confirmation', column_name_3='is_invitation', column_name_main='email_text')

Method trains 3 pipelines with the usage of data given in argument. Default option is the data from ‘data’ folder

train_multiclassifier_pipeline(path='../data/training_emails_multiclassifier.csv', classifier_option='MultinomialNB', column_name_train='email_type', column_name_main='email_text', estimator_1=None, estimator_2=None, estimator_3=None, voting_option='hard')

Method trains a pipeline that utilizes multiclassification.

view_3_stage_pipelines_accuracy()

Method displays the accuracy of the 3 pipelines.

view_multiclassifier_accuracy()

Method displays the accuracy of the multiclassifier.

Module contents