eric-chen-forward 0.0.6

Classifier for institution and scholar data

eric_chen_forward

To train the model:

from eric_chen_forward.model import Classifier

model = Classifier()

# option 1
# text files of labels and passages respectively, separated by newlines
model.train("labels_file_path", "passages_file_path")

# option 2
# csv file with a 'label' column and 'passage' column, the column names are hardcoded
model.train(csv_file="csv_file_path")

To use the saved model in code:

with open('model.pkl', 'rb') as f:
    model = pickle.load(f)

To run the classifier demo:

from eric_chen_forward import url_classifier_demo

url_classifier_demo.Demo('file path of model.pkl')