[{'role': 'system', 'content': "You are an expert in machine learning and your task is to classify code snippets into one of the following categories: ['helper_functions', 'load_data', 'data_exploration', 'data_preprocessing', 'evaluation', 'modelling', 'prediction', 'result_visualization', 'save_results', 'comment_only']."}, {'role': 'user', 'content': "Identify the single most appropriate category to which the provided code belongs. Provide your response as a single category label without any additional information, commentary, or personal opinions.\\nPython Code:teams_dict = teams.set_index(['Team_Id']).to_dict()\nprobframe['Team_1_Name'] = probframe.apply(lambda x:(teams_dict['Team_Name'][int(x.Team_1_id)]), axis =1)\nprobframe['Team_2_Name'] = probframe.apply(lambda x:(teams_dict['Team_Name'][int(x.Team_2_id)]), axis =1)\nprobframe.head()\nprobframe= probframe[['Season', 'Team_1_id', 'Team_2_id', 'Team_1_Name',\n'Team_2_Name', 'Pred']]\nprobframe.head()"}, {'role': 'assistant', 'content': 'data_preprocessing'}]