update dllib-sacred.py
Some checks failed
s444356-evaluation/pipeline/head This commit looks good
s444356-training/pipeline/head There was a failure building this commit

This commit is contained in:
Maciej Czajka 2022-05-09 21:08:39 +02:00
parent 452e1bedb5
commit 8e85f90e0a

View File

@ -249,7 +249,7 @@ epochs = int(sys.argv[1]) if len(sys.argv) > 20 else 20
#@ex.automain
#def my_main(epochs, _run):
def my_main(epochs)
def my_main(epochs):
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()