| | |
|---|
| | | ) -> List: |
|---|
| | | # make hotword list |
|---|
| | | hotwords, hotwords_length = self.proc_hotword(hotwords) |
|---|
| | | # import pdb; pdb.set_trace() |
|---|
| | | [bias_embed] = self.eb_infer(hotwords, hotwords_length) |
|---|
| | | # index from bias_embed |
|---|
| | | bias_embed = bias_embed.transpose(1, 0, 2) |
|---|
| | |
|---|
| | | return np.array(hotwords) |
|---|
| | | |
|---|
| | | hotword_int = [word_map(i) for i in hotwords] |
|---|
| | | # import pdb; pdb.set_trace() |
|---|
| | | |
|---|
| | | hotword_int.append(np.array([1])) |
|---|
| | | hotwords = pad_list(hotword_int, pad_value=0, max_len=10) |
|---|
| | | # import pdb; pdb.set_trace() |
|---|
| | | |
|---|
| | | return hotwords, hotwords_length |
|---|
| | | |
|---|
| | | def bb_infer( |
|---|
