From 5385c4bc3a5802e1caec979d0d3a6bc7af3e970f Mon Sep 17 00:00:00 2001 From: Adrien <adrien.payen@student.uclouvain.be> Date: Fri, 24 May 2024 21:31:51 +0200 Subject: [PATCH] update readme --- README.md | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index 366979d9..f0b21c60 100644 --- a/README.md +++ b/README.md @@ -110,21 +110,19 @@ Loading data, generating evaluation reports for different models, and saving exp ### Hackathon_make_predictions.ipynb -It defines a function make_hackathon_prediction that takes feature_method and regressor_method as input. +Define a function make_hackathon_prediction that takes feature_method and regressor_method as input. Inside this function: -- It loads the training data and converts it into the format suitable for Surprise. -- Trains a Content-Based model (ContentBased) on the training set using the specified feature and regressor methods. -- Makes predictions on the test set by loading the test data from a CSV file and converting it into records. +- Load the training data and converts it into the format suitable for Surprise. +- Train a Content-Based model (ContentBased) on the training set using the specified feature and regressor methods. +- Make predictions on the test set by loading the test data from a CSV file and converting it into records. Converts the predictions into a DataFrame and saves them as a CSV file. It then calls this function with specific parameters and prints the generated predictions. - - ### Dataset Organized under the data/test/ directory, which contains three subdirectories: - content: Contains movies.csv & tags.csv files. -- GitLab