chat-restaruacja/conllu_generator.py

16 lines
532 B
Python
Raw Normal View History

print("Script to automatically append data to data/dialog.conllu")
print("Start typing now. Press Ctrl+C to stop.")
while True:
with open("data/train_dialog.conllu", "a") as f:
text = input("Text: ")
act = input("Intent: ")
slots = text.split(" ")
f.write(
f"\n# text: {text}\n# intent: {act}\n# slots:\n"
)
for i, slot in enumerate(slots):
label = input(f"{i}/{slot} label: ")
f.write(f"{i+1}\t{slot}\t{act}\t{label}\n")
print("---")