[{'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:"""\nDON\'T MODIFY ANYTHING IN THIS CELL\n"""\nsave_model_path = \'./image_classification\'\nprint(\'Training...\')\nwith tf.Session() as sess:\nsess.run(tf.global_variables_initializer())\nfor epoch in range(epochs):\nn_batches = 5\nfor batch_i in range(1, n_batches + 1):\nfor batch_features, batch_labels in helper.load_preprocess_training_batch(batch_i, batch_size):\ntrain_neural_network(sess, optimizer, keep_probability, batch_features, batch_labels)\nprint(\'Epoch {:>2}, CIFAR-10 Batch {}:  \'.format(epoch + 1, batch_i), end=\'\')\nprint_stats(sess, batch_features, batch_labels, cost, accuracy)\nsaver = tf.train.Saver()\nsave_path = saver.save(sess, save_model_path)'}, {'role': 'assistant', 'content': 'modelling'}]