hw6.sh

orbxball · Jun 2, 2017 · 72b3bf0 · 72b3bf0
1 parent 3d6c92a
commit 72b3bf0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/hw6/hw6.sh b/hw6/hw6.sh
@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+
+# $1: the data directory including test.csv, users.csv, movies.csv
+# $2: prediction filename
+
+python3 test.py $1 $2
diff --git a/hw6/test.py b/hw6/test.py
@@ -57,7 +57,7 @@ def main(args):
  TEST_CSV = 'test.csv'
  USERS_CSV = 'users.csv'
  MOVIES_CSV = 'movies.csv'
- MODEL_WEIGHTS_FILE = 'weights.h5'
+ MODEL_WEIGHTS_FILE = 'weights_cf_bias.h5'
 
  DATA_DIR = args.data_dir
  TEST_CSV = os.path.join(DATA_DIR, TEST_CSV)