dropout=0.1; default dimension=15; epochs=1000

orbxball · Jun 1, 2017 · ef81619 · ef81619
1 parent 5000ed0
commit ef81619
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/hw6/Model.py b/hw6/Model.py
@@ -9,10 +9,12 @@ def build_cf_model(n_users, n_movies, dim):
  u_input = Input(shape=(1,))
  u = Embedding(n_users, dim, embeddings_regularizer=l2(1e-5))(u_input)
  u = Reshape((dim,))(u)
+ u = Dropout(0.1)(u)
 
  m_input = Input(shape=(1,))
  m = Embedding(n_movies, dim, embeddings_regularizer=l2(1e-5))(m_input)
  m = Reshape((dim,))(m)
+ m = Dropout(0.1)(m)
 
  u_bias = Embedding(n_users, 1, embeddings_regularizer=l2(1e-5))(u_input)
  u_bias = Reshape((1,))(u_bias)

diff --git a/hw6/train.py b/hw6/train.py
@@ -12,7 +12,7 @@ def parse_args():
  parser = argparse.ArgumentParser(description='HW6: Matrix Factorization')
  parser.add_argument('train', type=str)
  parser.add_argument('test', type=str)
- parser.add_argument('--dim', type=int, default=120)
+ parser.add_argument('--dim', type=int, default=15)
  return parser.parse_args()
 
 
@@ -49,7 +49,7 @@ def main(args):
 
  callbacks = [EarlyStopping('val_rmse', patience=2),
  ModelCheckpoint(MODEL_WEIGHTS_FILE, save_best_only=True)]
- history = model.fit([Users, Movies], Ratings, epochs=100, batch_size=256, validation_split=.1, verbose=1, callbacks=callbacks)
+ history = model.fit([Users, Movies], Ratings, epochs=1000, batch_size=256, validation_split=.1, verbose=1, callbacks=callbacks)
 
 
 if __name__ == '__main__':