diff --git a/README.md b/README.md index 366979d918014b2d74074003284f0258af5a76c1..f0b21c60a63fa23a565c5add64116223cd20dc1e 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.