import zipfile import io import requests import os from elliot.run import run_experiment url = "http://files.grouplens.org/datasets/movielens/ml-1m.zip" print(f"Getting Movielens 1Million from : {url} ..") response = requests.get(url) ml_1m_ratings = [] print("Extracting ratings.dat ..") with zipfile.ZipFile(io.BytesIO(response.content)) as zip_ref: for line in zip_ref.open("ml-1m/ratings.dat"): ml_1m_ratings.append(str(line, "utf-8").replace("::", "\t")) print("Printing ratings.tsv to data/cat_dbpedia_movielens_1m/ ..") os.makedirs("data/cat_dbpedia_movielens_1m", exist_ok=True) with open("data/cat_dbpedia_movielens_1m/dataset.tsv", "w") as f: f.writelines(ml_1m_ratings) print("Done! We are now starting the Elliot's experiment") run_experiment("config_files/basic_configuration.yml")
import zipfile import io import requests import os from elliot.run import run_experiment url = "http://files.grouplens.org/datasets/movielens/ml-1m.zip" print(f"Getting Movielens 1Million from : {url} ..") response = requests.get(url) ml_1m_ratings = [] print("Extracting ratings.dat ..") with zipfile.ZipFile(io.BytesIO(response.content)) as zip_ref: for line in zip_ref.open("ml-1m/ratings.dat"): ml_1m_ratings.append(str(line, "utf-8").replace("::", "\t")) print("Printing ratings.tsv to data/movielens_1m/ ..") os.makedirs("data/movielens_1m", exist_ok=True) with open("data/movielens_1m/dataset.tsv", "w") as f: f.writelines(ml_1m_ratings) print("Done! We are now starting the Elliot's experiment") run_experiment("config_files/advanced_configuration.yml")
from elliot.run import run_experiment import argparse parser = argparse.ArgumentParser(description="Run sample main.") parser.add_argument('--type_experiment', type=str, default='extract_attentive') parser.add_argument('--dataset', type=str, default='amazon_men') args = parser.parse_args() run_experiment("config_files/{0}_{1}.yml".format(args.type_experiment, args.dataset))
import zipfile import io import requests import os from elliot.run import run_experiment url = "http://files.grouplens.org/datasets/movielens/ml-100k.zip" print(f"Getting Movielens 100K from : {url} ..") response = requests.get(url) ml_100k_ratings = [] print("Extracting ratings ..") with zipfile.ZipFile(io.BytesIO(response.content)) as zip_ref: for line in zip_ref.open("ml-100k/u.data"): ml_100k_ratings.append(str(line, "utf-8")) print("Printing ratings.tsv to data/movielens_100k/ ..") os.makedirs("data/movielens_100k", exist_ok=True) with open("data/movielens_100k/dataset.tsv", "w") as f: f.writelines(ml_100k_ratings) print("Done! We are now starting the Elliot's experiment") run_experiment("config_files/sample_gan.yml")
import argparse from elliot.run import run_experiment parser = argparse.ArgumentParser() parser.add_argument('--model', type=str, default='most_pop', help='Model name') args = parser.parse_args() if args.model in ['most_pop', 'MF', 'NeuMF', 'ials', 'slim', 'easer', 'rp3beta', 'pure_svd']: run_experiment(f'''config_files/recsys_config_pinterest_{args.model}.yml''') else: raise ValueError('Model Selected is not available for this experiment')
import argparse from elliot.run import run_experiment parser = argparse.ArgumentParser() parser.add_argument('--dataset', type=str, default='movielens', help='dataset name in range [movilens, pintereset]') args = parser.parse_args() if args.dataset in ['movilens', 'pintereset']: run_experiment(f'''config_files/test_config_{args.dataset}.yml''') else: raise ValueError('Dataset selected is not available for this experiment')
from elliot.run import run_experiment run_experiment("config_files/facebook-books_kgflex.yml") run_experiment("config_files/facebook-books_baselines.yml") run_experiment("config_files/yahoo-movies_kgflex.yml") run_experiment("config_files/yahoo-movies_baselines.yml") run_experiment("config_files/movielens1m_kgflex.yml") run_experiment("config_files/movielens1m_baselines.yml") run_experiment("config_files/yahoo-movies_semantics-analysis.yml")
import argparse from elliot.run import run_experiment parser = argparse.ArgumentParser() parser.add_argument('--model', type=str, default='most_pop', help='Model name') args = parser.parse_args() if args.model in [ 'most_pop', 'MF', 'NeuMF', 'ials', 'slim', 'easer', 'rp3beta', 'pure_svd' ]: run_experiment( f'''config_files/recsys_config_movielens_{args.model}.yml''') else: raise ValueError('Model selected is not available for this experiment')
import zipfile import io import requests import os from elliot.run import run_experiment url = "http://files.grouplens.org/datasets/movielens/ml-1m.zip" print(f"Getting Movielens 1Million from : {url} ..") response = requests.get(url) ml_1m_ratings = [] print("Extracting ratings.dat ..") with zipfile.ZipFile(io.BytesIO(response.content)) as zip_ref: for line in zip_ref.open("ml-1m/ratings.dat"): ml_1m_ratings.append(str(line, "utf-8").replace("::", "\t")) print("Printing ratings.tsv to data/movielens_1m/ ..") os.makedirs("data/movielens_1m", exist_ok=True) with open("data/movielens_1m/dataset.tsv", "w") as f: f.writelines(ml_1m_ratings) print("Done! We are now starting the Elliot's experiment") run_experiment("config_files/sample_hello_world.yml")