def __call__(self, query_texts):
    df = pd.DataFrame()
    if self.use_bert:
        (preds, probs) = self.bert(query_texts)
        df['bert_labels'] = preds
        df['bert_probs'] = probs
    if self.use_ed:
        (preds, probs) = self.ed(query_texts, self.dictionary)
        df['diff_labels'] = [self.idx_mapping[x] for x in preds]
        df['diff_probs'] = probs
    if self.use_trie:
        (preds, probs) = self.trie(query_texts, self.dictionary)
        df['trie_labels'] = [self.idx_mapping[x] for x in preds]
        df['trie_probs'] = probs
    timestamps = retrieval.regex_timestamp(query_texts)
    df['timestamp'] = timestamps
    (preds, probs) = self.ensemble(df)
    return (preds, probs)
def __call__(self, query_texts):
    df = pd.DataFrame()
    if self.use_bert:
        (preds, probs) = self.bert(query_texts)
        df['bert_labels'] = preds
        df['bert_probs'] = probs
    if self.use_ed:
        (preds, probs) = self.ed(query_texts, self.dictionary)
        df['diff_labels'] = [self.idx_mapping[x] for x in preds]
        df['diff_probs'] = probs
    if self.use_trie:
        (preds, probs) = self.trie(query_texts, self.dictionary)
        df['trie_labels'] = [self.idx_mapping[x] for x in preds]
        df['trie_probs'] = probs
    timestamps = retrieval.regex_timestamp(query_texts)
    df['timestamp'] = timestamps
    (preds, probs) = self.ensemble(df)
    return (preds, probs)