challenging-america-word-gap-prediction/run.py

#!/usr/bin/python3
import sys


for line in sys.stdin:
    _, _, _, _, _, _, left_context, right_context = line.split("\t")
    
    previous_word = left_context.split()[-1]
    next_word = right_context.split()[0]
    
    if previous_word == "United" and next_word == "of":
        print("States:0.9 :0.1")
    elif previous_word == "used":
        print("to:0.4 it:0.3 as:0.2 :0.1")
    elif previous_word.lower() == "in":
        print("the:0.7 a:0.1 an:0.1 :0.1")
    elif previous_word.lower() == "i":
        print("am:0.3 was:0.3 have:0.3 :0.1")
    elif previous_word.lower() in ["he", "she", "it"]:
        print("is:0.3 was:0.3 has:0.3 :0.1")
    elif previous_word.lower() in "bring":
        print("something:0.3 it:0.3 them:0.3 :0.1")
    else:
        print("the:0.5 a:0.2 an:0.2 :0.1")