Open goodmami opened 7 years ago
Just to clarify, are you talking about the train
and test
functions in models.py or main.py? I'm assuming models.py
Actually I was referring to main.py in that we need a function to handle the lgid classify ...
command, but this might necessitate changes to models.py. As it is, models.py was just copied from https://github.com/xigt/classifier-common, so you might find it has parts missing or parts we don't use.
We have
train
andtest
commands defined, which train the doc-ID model and evaluate it, respectively. We also need aclassify
command that applies the model to new data, but doesn't evaluate it. In order to reduce redundant code, the current test() function could perhaps use the classify() function to get a prediction, then use the gold label for comparison.