def METHOD_NAME(data, tokenized_data, batch_size, pos_fraction=0.001, group_fraction=1.0):
	# data is the original data with all question ids and extra info. 
	# tokenized_data is the one after bert tokenization
	batch_list, group_id_list = _create_batches_ids(data, batch_size, pos_fraction, group_fraction)
	batch_matrix_list = _create_batch_matrix(group_id_list)
	# chaning the order in train_dataset so that when shuffle=False and DataLoader is used it gives the batches as we want.
	pl_tokenized_data = []
	for batch in batch_list:
		for idx in batch:
			pl_tokenized_data.append(tokenized_data[idx])
	pl_data_loader = DataLoader(pl_tokenized_data, batch_size=batch_size,shuffle=False, 
										batch_sampler=None, pin_memory=True)
	return pl_data_loader, batch_matrix_list