Python run_experiment Exemples

Exemple #1

Fichier : sample_basic.py Projet : danielemalitesta/elliot_expl

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")

Exemple #2

Fichier : sample_advanced.py Projet : merrafelice/elliot_eval

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")

Exemple #3

Fichier : sample_main.py Projet : danielemalitesta/elliot_expl

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))

Exemple #4

Fichier : sample_gan.py Projet : JavierSanzCruza/elliot

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")

Exemple #5

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')

Exemple #6

Fichier : evaluation.py Projet : sisinflab/Reenvisioning-the-comparison-between-Neural-Collaborative-Filtering-and-Matrix-Factorization

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')

Exemple #7

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")

Exemple #8

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')

Exemple #9

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")