INNER CODE UNIT · Python

Clause

abhijithneilabraham/tableQA · tableqa/clauses.py:11

class Clause:


    def __init__(self):
        self.distinct_types ={0: 'SELECT DISTINCT {} FROM {}', 1: 'SELECT MAX(DISTINCT {}) FROM {}', 2: 'SELECT MIN(DISTINCT {}) FROM {}',
                      3: 'SELECT COUNT(DISTINCT {}) FROM {}', 4: 'SELECT SUM(DISTINCT {}) FROM {}', 5: 'SELECT AVG(DISTINCT {}) FROM {}'}

        self.types = {0: 'SELECT {} FROM {}', 1: 'SELECT MAX({}) FROM {}', 2: 'SELECT MIN({}) FROM {}',
                      3: 'SELECT COUNT({}) FROM {}', 4: 'SELECT SUM({}) FROM {}', 5: 'SELECT AVG({}) FROM {}'}

    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))]

View source record →