Ejemplo n.º 1
0
    def __init__(self,
                 verbosity=1,
                 numLabels=3,
                 modelDir="ClassificationModelEndpoint",
                 unionSparsity=0.20):
        """
    Initializes the encoder as CioEncoder; requires a valid API key.
    """
        super(ClassificationModelEndpoint, self).__init__(verbosity=verbosity,
                                                          numLabels=numLabels,
                                                          modelDir=modelDir)

        root = os.path.dirname(os.path.realpath(__file__))
        self.encoder = CioEncoder(cacheDir=os.path.join(root, "CioCache"),
                                  unionSparsity=unionSparsity)
        self.compareEncoder = LanguageEncoder()

        self.n = self.encoder.n
        self.w = int((self.encoder.targetSparsity / 100) * self.n)

        self.categoryBitmaps = {}
        self.negatives = defaultdict(list)
        self.positives = defaultdict(list)