bilbo.components.svm package

Submodules

bilbo.components.svm.svm module

SVM

class bilbo.components.svm.svm.Svm(cfg_file, type_config='ini')

Bases: bilbo.components.component.Estimator

SVM class

evaluate(document)

Evaluate the model for the given data. All the data are split into 80/20% for the training / testing process

Parameters:document – document object

extract_xy(data)

fit(document, keep_on_doc=False)

generate_vocab_dict(document)

get_svm_data(document, keep_on_doc)

shape the svm features data for the svm :param document: document object of the document :returns: svm shaped data

predict(document)

tag the new data basde on a given model

Parameters:document – document object Returns:list of predictions

train(document)

Train the SVM model

Parameters:document – document object

transform(document, mode)

word_count(section)

words_iterator(section)

write_data_svm(data)

Module contents

SVM module (train / tag)