INNER CODE UNIT · Python

adapt

abhijithneilabraham/tableQA · tableqa/clauses.py:25

    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 →