コード例 #1
0
    def CreateLss(self):  #freq vec is the speaker tfidf
        Ns = self.prep.numSpeakers
        Lss = np.zeros((Ns, Ns))

        for i in range(0, Ns):
            if i + 1 in self.segm.cleanSpeakers[self.íter]:
                v1 = Help.CreateSpeakerVector(
                    i, self.segm.cleanSentences[self.íter],
                    self.segm.cleanSpeakers[self.íter], self.speakVec)
                for j in range(0, Ns):
                    if j + 1 in self.segm.cleanSpeakers[self.íter]:
                        v2 = Help.CreateSpeakerVector(
                            j, self.segm.cleanSentences[self.íter],
                            self.segm.cleanSpeakers[self.íter], self.speakVec)
                        if Help.NotValidCos(v1, v2):
                            v1, v2 = Help.ReshapeVec(v1, v2)
                        cos_dist = 1 - sp.distance.cosine(v1, v2)
                        if math.isnan(cos_dist):
                            Lss[i][j] = 0.
                        else:
                            Lss[i][j] = cos_dist

        return Lss
コード例 #2
0
 def CreateLus(self):
     Ns = self.prep.numSpeakers
     Lus = np.zeros((len(self.segm.cleanSentences[self.íter]), Ns))
     for i in range(0, len(self.segm.cleanSentences[self.íter])):
         v1 = Help.CreateSentenceVector(
             self.segm.cleanSentences[self.íter][i], self.freqVec,
             self.prep.singleWords)
         for j in range(0, Ns):
             if j + 1 in self.segm.cleanSpeakers[self.íter]:
                 v2 = Help.CreateSpeakerVector(
                     j, self.segm.cleanSentences[self.íter],
                     self.segm.cleanSpeakers[self.íter], self.speakVec)
                 if Help.NotValidCos(v1, v2):
                     v1, v2 = Help.ReshapeVec(v1, v2)
                 cos_dist = 1 - sp.distance.cosine(v1, v2)
                 if math.isnan(cos_dist):
                     Lus[i][j] = 0.
                 else:
                     Lus[i][j] = cos_dist
     return Lus