# -- Code Cell --
import pandas as pd

train_df = pd.read_csv('train_data.csv')
test_df = pd.read_csv('test_data.csv')
unlabeled_df = pd.read_csv('unlabeled_data.csv')
submission = pd.read_csv('sample_output.csv')

# -- Code Cell --
# TODO: Replace this with your actual predictions!
def solve1():
    return submission.loc[submission['subtaskID'] == 1, 'answer']

def solve2():
    return submission.loc[submission['subtaskID'] == 2, 'answer']

# -- Code Cell --
submission.to_csv('submission.csv', index=False)
submission.loc[submission['subtaskID'] == 2, 'answer'].value_counts()