Carlos-Francisco Méndez-Cruz / conditional-random-fields

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
98ff15c3 1 parent af5d0607

Conditional Random Fields

Inline Side-by-side
Showing 1 changed file with 4 additions and 4 deletions
......@@ -234,7 +234,7 @@ if __name__ == "__main__":
# crf = rs.best_estimator_
nameReport = options.trainingFile.replace('.txt', '.fStopWords_' + str(options.excludeStopWords) + '.fSymbols_' + str(
options.excludeSymbols) + '.txt')
with open(os.path.join(options.outputPath, "reports-l", "report_" + nameReport), mode="w") as oFile:
with open(os.path.join(options.outputPath, "reports-lp", "report_" + nameReport), mode="w") as oFile:
oFile.write("********** TRAINING **********\n")
oFile.write("Training file: " + options.trainingFile + '\n')
oFile.write('\n')
......@@ -256,14 +256,14 @@ if __name__ == "__main__":
print("*********************************")
name = options.trainingFile.replace('.txt', '.fStopWords_' + str(options.excludeStopWords) + '.fSymbols_' + str(
options.excludeSymbols) + '.txt')
with open(os.path.join(options.outputPath, "reports-l", "y_pred_" + name), "w") as oFile:
with open(os.path.join(options.outputPath, "reports-lp", "y_pred_" + name), "w") as oFile:
for y in y_pred:
oFile.write(str(y) + '\n')
print("*********************************")
name = options.trainingFile.replace('.txt', '.fStopWords_' + str(options.excludeStopWords) + '.fSymbols_' + str(
options.excludeSymbols) + '.txt')
with open(os.path.join(options.outputPath, "reports-l", "y_test_" + name), "w") as oFile:
with open(os.path.join(options.outputPath, "reports-lp", "y_test_" + name), "w") as oFile:
for y in y_test:
oFile.write(str(y) + '\n')
......@@ -272,7 +272,7 @@ if __name__ == "__main__":
# labels = list(crf.classes_)
# labels.remove('O')
with open(os.path.join(options.outputPath, "reports-l", "report_" + nameReport), mode="a") as oFile:
with open(os.path.join(options.outputPath, "reports-lp", "report_" + nameReport), mode="a") as oFile:
oFile.write('\n********** EVALUATION **********\n')
oFile.write("Flat F1: " + str(metrics.flat_f1_score(y_test, y_pred, average='weighted', labels=labels)))
oFile.write('\n')
......