diff --git a/data/small/recs/audrey_model.p b/data/small/recs/audrey_model.p new file mode 100644 index 0000000000000000000000000000000000000000..8de2366d7ac65ccdf03a61b846c222a8b7f86e87 Binary files /dev/null and b/data/small/recs/audrey_model.p differ diff --git a/library_audrey.csv b/library_audrey.csv new file mode 100644 index 0000000000000000000000000000000000000000..e5e68b75a49968384bb1094a3c64043e7c9b311d --- /dev/null +++ b/library_audrey.csv @@ -0,0 +1,201 @@ +movieId;title;genres;n_watched;wishlist;recent;top_10 +10;GoldenEye (1995);Action|Adventure|Thriller;0;0;0;0 +17;Sense and Sensibility (1995);Drama|Romance;0;1;0;0 +39;Clueless (1995);Comedy|Romance;0;1;0;0 +47;Seven (a.k.a. Se7en) (1995);Mystery|Thriller;0;0;0;0 +50;Usual Suspects, The (1995);Crime|Mystery|Thriller;0;0;0;0 +62;Mr. Holland's Opus (1995);Drama;0;0;0;0 +110;Braveheart (1995);Action|Drama|War;0;0;0;0 +150;Apollo 13 (1995);Adventure|Drama|IMAX;0;1;0;0 +153;Batman Forever (1995);Action|Adventure|Comedy|Crime;1;0;1;0 +165;Die Hard: With a Vengeance (1995);Action|Crime|Thriller;0;0;0;0 +185;Net, The (1995);Action|Crime|Thriller;1;0;1;0 +208;Waterworld (1995);Action|Adventure|Sci-Fi;0;0;0;0 +223;Clerks (1994);Comedy;0;1;0;0 +253;Interview with the Vampire: The Vampire Chronicles (1994);Drama|Horror;0;0;0;0 +292;Outbreak (1995);Action|Drama|Sci-Fi|Thriller;0;0;0;0 +296;Pulp Fiction (1994);Comedy|Crime|Drama|Thriller;0;0;0;0 +300;Quiz Show (1994);Drama;0;0;0;0 +339;While You Were Sleeping (1995);Comedy|Romance;0;1;0;0 +349;Clear and Present Danger (1994);Action|Crime|Drama|Thriller;0;0;0;0 +356;Forrest Gump (1994);Comedy|Drama|Romance|War;2;0;1;0 +357;Four Weddings and a Funeral (1994);Comedy|Romance;2;0;1;0 +364;Lion King, The (1994);Adventure|Animation|Children|Drama|Musical|IMAX;3;0;1;0 +367;Mask, The (1994);Action|Comedy|Crime|Fantasy;0;0;0;0 +377;Speed (1994);Action|Romance|Thriller;0;0;0;0 +454;Firm, The (1993);Drama|Thriller;1;0;1;0 +457;Fugitive, The (1993);Thriller;0;0;0;0 +480;Jurassic Park (1993);Action|Adventure|Sci-Fi|Thriller;1;0;1;0 +500;Mrs. Doubtfire (1993);Comedy|Drama;0;1;0;0 +508;Philadelphia (1993);Drama;0;1;0;0 +527;Schindler's List (1993);Drama|War;2;0;1;0 +539;Sleepless in Seattle (1993);Comedy|Drama|Romance;0;1;0;0 +551;Nightmare Before Christmas, The (1993);Animation|Children|Fantasy|Musical;0;0;0;0 +586;Home Alone (1990);Children|Comedy;0;0;0;0 +587;Ghost (1990);Comedy|Drama|Fantasy|Romance|Thriller;0;0;0;0 +588;Aladdin (1992);Adventure|Animation|Children|Comedy|Musical;3;0;1;0 +589;Terminator 2: Judgment Day (1991);Action|Sci-Fi;0;0;0;0 +590;Dances with Wolves (1990);Adventure|Drama|Western;2;0;1;1 +592;Batman (1989);Action|Crime|Thriller;1;0;1;0 +593;Silence of the Lambs, The (1991);Crime|Horror|Thriller;0;0;0;0 +318;Shawshank Redemption, The (1994);Crime|Drama;0;0;0;0 +595;Beauty and the Beast (1991);Animation|Children|Fantasy|Musical|Romance|IMAX;2;0;1;0 +736;Twister (1996);Action|Adventure|Romance|Thriller;0;1;0;0 +778;Trainspotting (1996);Comedy|Crime|Drama;0;0;0;0 +1.197;Princess Bride, The (1987);Action|Adventure|Comedy|Fantasy|Romance;0;1;0;0 +1.210;Star Wars: Episode VI - Return of the Jedi (1983);Action|Adventure|Sci-Fi;1;0;1;0 +1.580;Men in Black (a.k.a. MIB) (1997);Action|Comedy|Sci-Fi;1;0;1;0 +1.721;Titanic (1997);Drama|Romance;6;0;1;1 +2.028;Saving Private Ryan (1998);Action|Drama|War;2;0;1;0 +2.716;Ghostbusters (a.k.a. Ghost Busters) (1984);Action|Comedy|Sci-Fi;1;0;1;0 +2.762;Sixth Sense, The (1999);Drama|Horror|Mystery;0;0;0;0 +2.858;American Beauty (1999);Drama|Romance;0;1;0;0 +2.959;Fight Club (1999);Action|Crime|Drama|Thriller;0;1;0;0 +5.349;Spider-Man (2002);Action|Adventure|Sci-Fi|Thriller;1;0;1;0 +6.377;Finding Nemo (2003);Adventure|Animation|Children|Comedy;1;0;1;0 +7.153;Lord of the Rings: The Return of the King, The (2003);Action|Adventure|Drama|Fantasy;0;0;0;0 +7.361;Eternal Sunshine of the Spotless Mind (2004);Drama|Romance|Sci-Fi;0;1;0;0 +8.636;Spider-Man 2 (2004);Action|Adventure|Sci-Fi|IMAX;0;0;0;0 +58.559;Dark Knight, The (2008);Action|Crime|Drama|IMAX;0;0;0;0 +34;Babe (1995);Children|Drama;0;0;0;0 +141;Birdcage, The (1996);Comedy;0;0;0;0 +260;Star Wars: Episode IV - A New Hope (1977);Action|Adventure|Sci-Fi;1;0;1;0 +329;Star Trek: Generations (1994);Adventure|Drama|Sci-Fi;0;0;0;0 +380;True Lies (1994);Action|Adventure|Comedy|Romance|Thriller;0;1;0;0 +434;Cliffhanger (1993);Action|Adventure|Thriller;0;0;0;0 +440;Dave (1993);Comedy|Romance;0;1;0;0 +442;Demolition Man (1993);Action|Adventure|Sci-Fi;0;0;0;0 +541;Blade Runner (1982);Action|Sci-Fi|Thriller;0;0;0;0 +858;Godfather, The (1972);Crime|Drama;0;0;0;0 +919;Wizard of Oz, The (1939);Adventure|Children|Fantasy|Musical;0;0;0;0 +1.036;Die Hard (1988);Action|Crime|Thriller;0;0;0;0 +1.073;Willy Wonka & the Chocolate Factory (1971);Children|Comedy|Fantasy|Musical;4;0;1;1 +1.079;Fish Called Wanda, A (1988);Comedy|Crime;0;0;0;0 +1.089;Reservoir Dogs (1992);Crime|Mystery|Thriller;0;0;0;0 +1.097;E.T. the Extra-Terrestrial (1982);Children|Drama|Sci-Fi;2;0;1;0 +1.136;Monty Python and the Holy Grail (1975);Adventure|Comedy|Fantasy;0;0;0;0 +1.196;Star Wars: Episode V - The Empire Strikes Back (1980);Action|Adventure|Sci-Fi;1;0;1;0 +1.198;Raiders of the Lost Ark (Indiana Jones and the Raiders of the Lost Ark) (1981);Action|Adventure;1;0;1;0 +1.200;Aliens (1986);Action|Adventure|Horror|Sci-Fi;0;0;0;0 +1.206;Clockwork Orange, A (1971);Crime|Drama|Sci-Fi|Thriller;0;0;0;0 +1.208;Apocalypse Now (1979);Action|Drama|War;0;0;0;0 +1.213;Goodfellas (1990);Crime|Drama;0;0;0;0 +1.214;Alien (1979);Horror|Sci-Fi;0;0;0;0 +1.220;Blues Brothers, The (1980);Action|Comedy|Musical;0;0;0;0 +1.222;Full Metal Jacket (1987);Drama|War;0;0;0;0 +1.225;Amadeus (1984);Drama;0;0;0;0 +1.240;Terminator, The (1984);Action|Sci-Fi|Thriller;0;0;0;0 +1.258;Shining, The (1980);Horror;0;0;0;0 +1.259;Stand by Me (1986);Adventure|Drama;0;0;0;0 +1.265;Groundhog Day (1993);Comedy|Fantasy|Romance;0;0;0;0 +1.270;Back to the Future (1985);Adventure|Comedy|Sci-Fi;1;0;1;0 +1.291;Indiana Jones and the Last Crusade (1989);Action|Adventure;1;0;1;0 +1.307;When Harry Met Sally... (1989);Comedy|Romance;1;0;1;0 +1.380;Grease (1978);Comedy|Musical|Romance;3;0;1;0 +1.387;Jaws (1975);Action|Horror;0;0;0;0 +1.917;Armageddon (1998);Action|Romance|Sci-Fi|Thriller;0;0;0;0 +1.961;Rain Man (1988);Drama;0;0;0;0 +1.968;Breakfast Club, The (1985);Comedy|Drama;0;1;0;0 +2.000;Lethal Weapon (1987);Action|Comedy|Crime|Drama;0;0;0;0 +2.115;Indiana Jones and the Temple of Doom (1984);Action|Adventure|Fantasy;0;0;0;0 +2.174;Beetlejuice (1988);Comedy|Fantasy;0;0;0;0 +2.628;Star Wars: Episode I - The Phantom Menace (1999);Action|Adventure|Sci-Fi;0;0;0;0 +2.683;Austin Powers: The Spy Who Shagged Me (1999);Action|Adventure|Comedy;0;1;0;0 +2.791;Airplane! (1980);Comedy;0;1;0;0 +2.797;Big (1988);Comedy|Drama|Fantasy|Romance;0;0;0;0 +2.916;Total Recall (1990);Action|Adventure|Sci-Fi|Thriller;0;0;0;0 +2.918;Ferris Bueller's Day Off (1986);Comedy;0;0;0;0 +2.987;Who Framed Roger Rabbit? (1988);Adventure|Animation|Children|Comedy|Crime|Fantasy|Mystery;0;0;0;0 +104;Happy Gilmore (1996);Comedy;1;0;1;0 +231;Dumb & Dumber (Dumb and Dumber) (1994);Adventure|Comedy;0;0;0;0 +344;Ace Ventura: Pet Detective (1994);Comedy;0;0;0;0 +597;Pretty Woman (1990);Comedy|Romance;3;0;1;1 +1.193;One Flew Over the Cuckoo's Nest (1975);Drama;0;0;0;0 +1.221;Godfather: Part II, The (1974);Crime|Drama;0;0;0;0 +1.247;Graduate, The (1967);Comedy|Drama|Romance;0;1;0;0 +1.393;Jerry Maguire (1996);Drama|Romance;0;0;0;0 +1.682;Truman Show, The (1998);Comedy|Drama|Sci-Fi;0;0;0;0 +1.784;As Good as It Gets (1997);Comedy|Drama|Romance;0;1;0;0 +1.923;There's Something About Mary (1998);Comedy|Romance;0;1;0;0 +2.355;Bug's Life, A (1998);Adventure|Animation|Children|Comedy;0;0;0;0 +2.502;Office Space (1999);Comedy|Crime;0;0;0;0 +2.706;American Pie (1999);Comedy|Romance;0;0;0;0 +2.997;Being John Malkovich (1999);Comedy|Drama|Fantasy;0;0;0;0 +3.114;Toy Story 2 (1999);Adventure|Animation|Children|Comedy|Fantasy;1;0;1;0 +3.408;Erin Brockovich (2000);Drama;0;0;0;0 +3.897;Almost Famous (2000);Drama;1;0;1;0 +3.948;Meet the Parents (2000);Comedy;1;0;1;0 +4.022;Cast Away (2000);Drama;0;0;0;0 +4.306;Shrek (2001);Adventure|Animation|Children|Comedy|Fantasy|Romance;1;0;1;0 +4.963;Ocean's Eleven (2001);Crime|Thriller;3;0;1;1 +4.995;Beautiful Mind, A (2001);Drama|Romance;1;0;1;1 +6.711;Lost in Translation (2003);Comedy|Drama|Romance;0;1;0;0 +111;Taxi Driver (1976);Crime|Drama|Thriller;0;0;0;0 +293;Léon: The Professional (a.k.a. The Professional) (Léon) (1994);Action|Crime|Drama|Thriller;0;0;0;0 +2.571;Matrix, The (1999);Action|Sci-Fi|Thriller;1;0;1;0 +3.751;Chicken Run (2000);Animation|Children|Comedy;0;0;0;0 +5.952;Lord of the Rings: The Two Towers, The (2002);Adventure|Fantasy;0;0;0;0 +8.368;Harry Potter and the Prisoner of Azkaban (2004);Adventure|Fantasy|IMAX;3;0;1;1 +1;Toy Story (1995);Adventure|Animation|Children|Comedy|Fantasy;1;0;1;0 +21;Get Shorty (1995);Comedy|Crime|Thriller;0;0;0;0 +316;Stargate (1994);Action|Adventure|Sci-Fi;0;0;0;0 +780;Independence Day (a.k.a. ID4) (1996);Action|Adventure|Sci-Fi|Thriller;0;0;0;0 +924;2001: A Space Odyssey (1968);Adventure|Drama|Sci-Fi;0;0;0;0 +1.080;Monty Python's Life of Brian (1979);Comedy;0;0;0;0 +32;Twelve Monkeys (a.k.a. 12 Monkeys) (1995);Mystery|Sci-Fi|Thriller;0;0;0;0 +1.617;L.A. Confidential (1997);Crime|Film-Noir|Mystery|Thriller;0;0;0;0 +1.704;Good Will Hunting (1997);Drama|Romance;0;1;0;0 +2.324;Life Is Beautiful (La Vita è bella) (1997);Comedy|Drama|Romance|War;1;0;1;0 +2.329;American History X (1998);Crime|Drama;0;0;0;0 +3.147;Green Mile, The (1999);Crime|Drama;0;0;0;0 +3.578;Gladiator (2000);Action|Adventure|Drama;1;0;1;0 +3.996;Crouching Tiger, Hidden Dragon (Wo hu cang long) (2000);Action|Drama|Romance;0;0;0;0 +4.011;Snatch (2000);Comedy|Crime|Thriller;0;0;0;0 +4.226;Memento (2000);Mystery|Thriller;0;0;0;0 +4.886;Monsters, Inc. (2001);Adventure|Animation|Children|Comedy|Fantasy;2;0;1;0 +4.896;Harry Potter and the Sorcerer's Stone (a.k.a. Harry Potter and the Philosopher's Stone) (2001);Adventure|Children|Fantasy;3;0;1;0 +4.973;Amelie (Fabuleux destin d'Amélie Poulain, Le) (2001);Comedy|Romance;1;0;1;0 +4.993;Lord of the Rings: The Fellowship of the Ring, The (2001);Adventure|Fantasy;1;0;1;0 +5.445;Minority Report (2002);Action|Crime|Mystery|Sci-Fi|Thriller;0;0;0;0 +5.989;Catch Me If You Can (2002);Crime|Drama;0;1;0;0 +6.874;Kill Bill: Vol. 1 (2003);Action|Crime|Thriller;0;0;0;0 +7.438;Kill Bill: Vol. 2 (2004);Action|Drama|Thriller;0;0;0;0 +33.794;Batman Begins (2005);Action|Crime|IMAX;1;0;1;0 +36;Dead Man Walking (1995);Crime|Drama;1;0;1;0 +608;Fargo (1996);Comedy|Crime|Drama|Thriller;0;1;0;0 +733;Rock, The (1996);Action|Adventure|Thriller;1;0;1;0 +1.584;Contact (1997);Drama|Sci-Fi;0;0;0;0 +2.291;Edward Scissorhands (1990);Drama|Fantasy|Romance;0;0;0;0 +2.396;Shakespeare in Love (1998);Comedy|Drama|Romance;0;1;0;0 +923;Citizen Kane (1941);Drama|Mystery;0;0;0;0 +48.516;Departed, The (2006);Crime|Drama|Thriller;0;0;0;0 +79.132;Inception (2010);Action|Crime|Drama|Mystery|Sci-Fi|Thriller|IMAX;2;0;1;1 +1.732;Big Lebowski, The (1998);Comedy|Crime;0;0;0;0 +3.793;X-Men (2000);Action|Adventure|Sci-Fi;1;0;1;0 +4.878;Donnie Darko (2001);Drama|Mystery|Sci-Fi|Thriller;0;0;0;0 +2;Jumanji (1995);Adventure|Children|Fantasy;1;0;1;0 +6;Heat (1995);Action|Crime|Thriller;0;0;0;0 +16;Casino (1995);Crime|Drama;1;0;1;0 +19;Ace Ventura: When Nature Calls (1995);Comedy;0;0;0;0 +25;Leaving Las Vegas (1995);Drama|Romance;0;1;0;0 +95;Broken Arrow (1996);Action|Adventure|Thriller;1;0;1;0 +288;Natural Born Killers (1994);Action|Crime|Thriller;0;0;0;0 +648;Mission: Impossible (1996);Action|Adventure|Mystery|Thriller;2;0;1;1 +750;Dr. Strangelove or: How I Learned to Stop Worrying and Love the Bomb (1964);Comedy|War;0;1;0;0 +904;Rear Window (1954);Mystery|Thriller;0;0;0;0 +908;North by Northwest (1959);Action|Adventure|Mystery|Romance|Thriller;0;0;0;0 +912;Casablanca (1942);Drama|Romance;0;1;0;0 +1.246;Dead Poets Society (1989);Drama;2;0;1;1 +1.517;Austin Powers: International Man of Mystery (1997);Action|Adventure|Comedy;1;0;1;0 +1.527;Fifth Element, The (1997);Action|Adventure|Comedy|Sci-Fi;0;0;0;0 +1.610;Hunt for Red October, The (1990);Action|Adventure|Thriller;0;0;0;0 +1.653;Gattaca (1997);Drama|Sci-Fi|Thriller;0;0;0;0 +2.011;Back to the Future Part II (1989);Adventure|Comedy|Sci-Fi;1;0;1;0 +2.012;Back to the Future Part III (1990);Adventure|Comedy|Sci-Fi|Western;1;0;1;0 +2.710;Blair Witch Project, The (1999);Drama|Horror|Thriller;0;0;0;0 +3.481;High Fidelity (2000);Comedy|Drama|Romance;1;0;1;0 +4.027;O Brother, Where Art Thou? (2000);Adventure|Comedy|Crime;0;0;0;0 +5.418;Bourne Identity, The (2002);Action|Mystery|Thriller;0;0;0;0 +6.539;Pirates of the Caribbean: The Curse of the Black Pearl (2003);Action|Adventure|Comedy|Fantasy;2;0;1;0 +8.961;Incredibles, The (2004);Action|Adventure|Animation|Children|Comedy;1;0;1;0 +337;What's Eating Gilbert Grape (1993);Drama;0;1;0;0 \ No newline at end of file diff --git a/recommender_building.py b/recommender_building.py index 973fd8bca6db8bcd10e013b86b2cb3c96492e76e..0401b88fc6774f5b391132fff39ed7fb18e40219 100644 --- a/recommender_building.py +++ b/recommender_building.py @@ -72,7 +72,7 @@ def train_and_save_model(training_data, model_name): if __name__ == "__main__": # Path to your library file (e.g., library_adrien.csv) - library_file = 'library_adrien.csv' + library_file = 'library_audrey.csv' # Path to MovieLens ratings dataset file (e.g., ratings.csv) ratings_file = 'data/small/evidence/ratings.csv' @@ -84,5 +84,5 @@ if __name__ == "__main__": training_data = append_and_prepare_training_set(implicit_ratings_df, ratings_file) # Train and save model - model_name = 'adrien_model' + model_name = 'audrey_model' train_and_save_model(training_data, model_name)