INNER CODE UNIT · Python

get_embeddings

abhijithneilabraham/tableQA · tableqa/clauses.py:21

    def get_embeddings(self, x):
        embeddings = embed(x)
        return asarray(embeddings)

    def adapt(self, q, inttype=False, summable=False,distinct=False):
        emb = self.get_embeddings([q])
        if distinct:
            self.clause = self.distinct_types[argmax(model.predict(emb))]
        else:
            self.clause = self.types[argmax(model.predict(emb))]

        if summable and inttype and "COUNT" in self.clause:
            self.clause = '''SELECT SUM({}) FROM {}'''

        return self.clause

    

View source record →