-
Notifications
You must be signed in to change notification settings - Fork 49
/
sample_advanced.py
26 lines (19 loc) · 785 Bytes
/
sample_advanced.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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")