src.models package

Submodules

src.models.model module

class src.models.model.Model(clf)

Bases: object

Class represents the model used in classification

build_pipeline()

Method builds the pipeline with TfidfVectorizer and chosen classifier.

count_accuracy()

Method returns the accuracy of the model.

get_X()

Method returns the X data.

get_X_test()

Method returns the X test data.

get_X_train()

Method returns the X training data.

get_clf()

Method returns the classifier.

get_pipeline()

Method returns the pipeline.

get_random_state()

Method returns the random state.

get_test_size()

Method returns the test size.

get_y()

Method returns the y data.

get_y_pred()

Method returns the y_pred.

get_y_test()

Method returns the y test data.

get_y_train()

Method returns the y training data.

set_X(X)

Method sets the X data.

set_X_test(X_test)

Method sets the X test data.

set_X_train(X_train)

Method sets the X training data.

set_clf(clf)

Method sets the classifier.

set_random_state(random_state)

Method sets the random state.

set_test_size(test_size)

Method sets the test size.

set_y(y)

Method sets the y data.

set_y_pred(y_pred)

Method sets the y_pred.

set_y_test(y_test)

Method sets the y test data.

set_y_train(y_train)

Method sets the y training data.

train()

Method trains the model.

Module contents