[{'role': 'system', 'content': "You are an expert in machine learning and your task is to classify the code snippets into categories from the provided list of categories. List of categories: ['helper_functions', 'load_data', 'data_exploration', 'data_preprocessing', 'evaluation', 'modelling', 'prediction', 'result_visualization', 'save_results', 'comment_only']"}, {'role': 'user', 'content': "Identify to which category the provided code belongs and assign the code snippet to at least 1 but up to 5 categories based on the probabilities, ignore if the probability of the category is negligible. Strictly provide your response as a list. Avoid providing any additional information, commentary, or personal opinions.\\nPython Code:figure(1)\nfig = plt.figure(figsize=(16,10), dpi=dpi)\nplt.subplot(221)\ndata   = Kt_A\nfield  = np.ma.masked_array(data, mask=[data==0.])\ndepths = zi_A\ndeltaz = deltaz_A\nCS = plt.pcolormesh(time, -depths, field.T, vmin=0.0, vmax=0.14)\nplt.colorbar()\nplt.ylim((ymin,0))\nplt.ylabel('z (m)',fontsize=24)\nplt.title(r'MOM6 KPP $\\kappa_{\\Theta}$ ($m^2/s$) with $\\Delta z$ ='+deltaz,fontsize=24);\nplot = fig.add_subplot(221)\nplot.tick_params(axis='both', which='major', labelsize=18)\nplot.tick_params(axis='both', which='minor', labelsize=18)\nplt.subplot(222)\ndata   = Kt_B\nfield  = np.ma.masked_array(data, mask=[data==0.])\ndepths = zi_B\ndeltaz = deltaz_B\nCS = plt.pcolormesh(time, -depths, field.T, vmin=0.0, vmax=0.14)\nplt.colorbar()\nplt.ylim((ymin,0))\nplt.ylabel('z (m)',fontsize=24)\nplt.title(r'MOM6 KPP $\\kappa_{\\Theta}$ ($m^2/s$) with $\\Delta z$ ='+deltaz,fontsize=24);\nplot = fig.add_subplot(222)\nplot.tick_params(axis='both', which='major', labelsize=18)\nplot.tick_params(axis='both', which='minor', labelsize=18)\nplt.subplot(223)\ndata   = Kt_C\nfield  = np.ma.masked_array(data, mask=[data==0.])\ndepths = zi_C\ndeltaz = deltaz_C\nCS = plt.pcolormesh(time, -depths, field.T, vmin=0.0, vmax=0.14)\nplt.colorbar()\nplt.ylim((ymin,0))\nplt.ylabel('z (m)',fontsize=24)\nplt.title(r'MOM6 KPP $\\kappa_{\\Theta}$ ($m^2/s$) with $\\Delta z$ ='+deltaz,fontsize=24)\nplt.xlabel('Time (days)',fontsize=24);\nplot = fig.add_subplot(223)\nplot.tick_params(axis='both', which='major', labelsize=18)\nplot.tick_params(axis='both', which='minor', labelsize=18)\nplt.subplot(224)\ndata   = Kt_D\nfield  = np.ma.masked_array(data, mask=[data==0.])\ndepths = zi_D\ndeltaz = deltaz_D\nCS = plt.pcolormesh(time, -depths, field.T, vmin=0.0, vmax=0.14)\nplt.colorbar()\nplt.ylim((ymin,0))\nplt.ylabel('z (m)',fontsize=24)\nplt.title(r'MOM6 KPP $\\kappa_{\\Theta}$ ($m^2/s$) with $\\Delta z$ ='+deltaz,fontsize=24)\nplt.xlabel('Time (days)',fontsize=24);\nplot = fig.add_subplot(224)\nplot.tick_params(axis='both', which='major', labelsize=18)\nplot.tick_params(axis='both', which='minor', labelsize=18)\nplt.tight_layout()\nfname = expt_name+'_MOM6_KPP_diffusivity.png'\nfig.savefig(fname,dpi=dpi);"}, {'role': 'assistant', 'content': "['data_exploration']"}]