(self, trie, bos_token_id)

(log, color, highlight=True)

(self, prefix_sequence: List[int])

(self, sequence: List[int])

(self, args)

(self, data)

(self, data, prompt)

(self, data, prompt)

(self, xs, labels, use_sk=False)

(sequence: List[int], trie_dict: Dict)

(self, epoch, collision_rate=1, ckpt_file=None)

(distances)

(self, data, n_clusters=10)

(self)

(self, data)

(args)

(args)

(parser)

(parser)

(parser)

(parser)

(seed)

(seed)

(distances, epsilon, sinkhorn_iterations)

(self, quantized_rep, encoded_rep)

(self, args)

(self)

(self, loss)

(self)

(self)

(self, epoch_idx, s_time, e_time, loss, recon_loss, cf_loss)

( prefix_sequence: List[int], trie_dict: Dict, append_trie=None, bos_token_id:

(self, data, prompt)

(self, data, prompt)

(self, data, prompt)

(self, data, prompt)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self, train_data, epoch_idx)

(self, valid_data)

(activation_name="relu", emb_dim=None)

(all_indices_str)

(raw_text)

(self, out, quant_loss, emb_idx, dense_out, xs=None)

(data, n_clusters=10)

(self, data, n_clusters=10)

(self, x_q, indices, indices_cluster, indices_list)

(self, x, x_q, indices, labels)

(dir_path)

(dir_path)

(dir_path)

(args, item_text_list, tokenizer, model, word_drop_ratio=-1)

(item2feature, features)

(self)

(self)

(all_indices_str)

(all_indices_str)

(topk_results, metrics)

(topk_results, metrics)

(self)

(self, tokenizer)

(predictions, scores, targets, k, all_items=None)

(predictions, scores, targets, k, all_items=None)

(topk_results, k)

(topk_results, k)

(args)

(args)

(file)

(model_path='bert-base-uncased')

(args)

(topk_results, k)

()

()

()

(parser)