update dllib-sacred.py
All checks were successful
s444356-evaluation/pipeline/head This commit looks good
s444356-training/pipeline/head This commit looks good

This commit is contained in:
Maciej Czajka 2022-05-04 17:13:29 +02:00
parent 949429b0f5
commit d1d8e4c704

View File

@ -244,10 +244,10 @@ def remove_list(games):
@ex.automain
def my_main(epochs, _run):
platform = pd.read_csv('all_games_train.csv', sep=',', usecols=[1], header=None).values.tolist()
release_date = pd.read_csv('all_games_train.csv', sep=',', usecols=[2], header=None).values.tolist()
meta_score = pd.read_csv('all_games_train.csv', sep=',', usecols=[4], header=None).values.tolist()
user_review = pd.read_csv('all_games_train.csv', sep=',', usecols=[5], header=None).values.tolist()
platform = pd.read_csv('all_games.train.csv', sep=',', usecols=[1], header=None).values.tolist()
release_date = pd.read_csv('all_games.train.csv', sep=',', usecols=[2], header=None).values.tolist()
meta_score = pd.read_csv('all_games.train.csv', sep=',', usecols=[4], header=None).values.tolist()
user_review = pd.read_csv('all_games.train.csv', sep=',', usecols=[5], header=None).values.tolist()
games_train = {'platform': platform,
'release_date': release_date,