validation_labels, validation_asins, validation_reviewerIDs = pre_work.GenerateLabelEncoding( USER, (num_of_reviews + num_of_rating) + idx, 1, itemObj, userObj) _batch_labels, _asins, _reviewerIDs = pre_work.GenerateBatchLabelCandidate( validation_labels, validation_asins, validation_reviewerIDs, batch_size) validate_batch_labels.append(_batch_labels) validate_asins.append(_asins) validate_reviewerIDs.append(_reviewerIDs) #%% Generate training batches if (trainOption or validationOption): # Get training asin batches training_batches = pre_work.GenerateTrainingBatches( USER, itemObj, num_of_reviews=num_of_reviews, batch_size=batch_size) # %% if (trainOption): Train(voc, selectTable, training_batches, candidate_asins, candidate_reviewerIDs, training_batch_labels, directory, TrainEpoch=trainingEpoch, isStoreModel=True, WriteTrainLoss=True, store_every=2, batch_size=batch_size)
validation_labels, validation_asins, validation_reviewerIDs, batch_size) validate_batch_labels.append(_batch_labels) validate_asins.append(_asins) validate_reviewerIDs.append(_reviewerIDs) else: validate_batch_labels = None validate_asins = None validate_reviewerIDs = None # Generate training batches if (trainOption): training_batches, training_asin_batches = pre_work.GenerateTrainingBatches( USER, itemObj, voc, num_of_reviews=num_of_reviews, batch_size=batch_size) if (trainOption): Train(voc, selectTable, training_batches, training_asin_batches, candidate_asins, candidate_reviewerIDs, training_batch_labels, validate_batch_labels, validate_asins, validate_reviewerIDs, directory,