Base_model_ challenge.py

challenge_train_preds = []
challenge_dev_preds = []
challenge_test_preds = []

for i in range(len(df_challenge_train)):
    if '(1)' in df_challenge_train['only_answers'].iloc[i]:
        challenge_train_preds.append(random.choice(num_list))
    else:
        if '(E)' in df_challenge_train['only_answers'].iloc[i]:
            challenge_train_preds.append(random.choice(char_list_2))
        else:
            challenge_train_preds.append(random.choice(char_list_1))

for i in range(len(df_challenge_dev)):
    if '(1)' in df_challenge_dev['only_answers'].iloc[i]:
        challenge_dev_preds.append(random.choice(num_list))
    else:
        if '(E)' in df_challenge_dev['only_answers'].iloc[i]:
            challenge_dev_preds.append(random.choice(char_list_2))
        else:
            challenge_dev_preds.append(random.choice(char_list_1))

for i in range(len(df_challenge_test)):
    if '(1)' in df_challenge_test['only_answers'].iloc[i]:
        challenge_test_preds.append(random.choice(num_list))
    else:
        if '(E)' in df_challenge_test['only_answers'].iloc[i]:
            challenge_test_preds.append(random.choice(char_list_2))
        else:
            challenge_test_preds.append(random.choice(char_list_1))

ok