[{'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:def predict(classifier: sklearn.linear_model.LogisticRegression, indices: List[int]) -> numpy.ndarray:\n"""Predicts probabilities for a set of IR objects.\nArguments\n---------\nclassifier\nTrained classifier.\nindices\nList of IR indices to predict probability of.\nReturns\n-------\nnumpy.ndarray\n(N,) NumPy array of predicted probabilities.\n"""\nwith h5py.File(TRAINING_PATH, \'r\') as tr:\nfeatures = numpy.nan_to_num(tr[\'features\'].value[indices])\nreturn classifier.predict_proba(features)[:, 1]'}, {'role': 'assistant', 'content': "['load_data', 'modelling', 'prediction']"}]