def __init__(self, k=1): # phrase detector self.phrase_detector = PmiPhraseDetector(RawSentenceStream()) # number converter self.tokenizer = RawTokenizer() # build model self.model = W2Vmodel(PhraseSentenceStream(self.phrase_detector)) # parameters self.p = 0.8 self.k = k
def __init__(self, alpha=6.0): # phrase detector self.phrase_detector = PmiPhraseDetector(RawSentenceStream()) # number converter self.tokenizer = RawTokenizer() # build model self.w2v = W2Vmodel(PhraseSentenceStream(self.phrase_detector)) self.tfidf = TFIDFmodel() # parameters self.alpha = alpha self.k = 3 self.p = 0.80