datahandler = DataHandler(dataset_path, BATCHSIZE, MAX_SESSION_REPRESENTATIONS, 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