Run autopep8
This commit is contained in:
parent
d2d308a78a
commit
0238b238d2
@ -42,11 +42,13 @@ def predict(model, in_file, out_file):
|
||||
prediction = model.predict(lines)
|
||||
np.savetxt(out_file, prediction, fmt='%d')
|
||||
|
||||
|
||||
def main():
|
||||
#stop_words = set(stopwords.words('english'))
|
||||
model = train_model("train/in.tsv", "train/expected.tsv")
|
||||
predict(model, "dev-0/in.tsv", "dev-0/out.tsv")
|
||||
predict(model, "test-A/in.tsv", "test-A/out.tsv")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
Loading…
Reference in New Issue
Block a user