Ejemplo n.º 1
0
                          dims["INTRA_HIDDEN"], dims["TIME_RESOLUTION"],
                          min_time)
dims["N_ITEMS"] = datahandler.get_num_items()
N_SESSIONS = datahandler.get_num_training_sessions()
dims["N_USERS"] = datahandler.get_num_users()

# TODO: Initialize tester
tester = Tester("Log")
model = DynamicRecModel(dims, dropout, params, datahandler, tester,
                        time_threshold)

# setting up for training
epoch_nr = 0
start_time = time.time()
num_training_batches = datahandler.get_num_training_batches()
num_test_batches = datahandler.get_num_test_batches()
epoch_loss = 0

# start training
while epoch_nr < MAX_EPOCHS:
    with open(txt_log_name, 'a') as txt_file:
        txt_file.write("Starting epoch #" + str(epoch_nr) + "\n")
    start_time_epoch = time.time()

    # reset the datahandler and get first training batch
    datahandler.reset_user_batch_data_train()
    datahandler.reset_user_session_representations()
    items, item_targets, session_lengths, session_reps, session_rep_lengths, user_list, sess_time_reps, time_targets, first_rec_targets = datahandler.get_next_train_batch(
    )
    batch_nr = 0