Commit 79f2895

committed

bug fix : word length

1 parent 93ccb78 commit 79f2895Copy full SHA for 79f2895

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ def generate_sentences_batch(self):`
`40`	`40`	`labels_batch = []`
`41`	`41`	`for i, words in enumerate(self.words_index):`
`42`	`42`	`if len(words) < self.skip_window:`
`43`		`- return`
	`43`	`+ continue`
`44`	`44`	`extend_words = [1] * self.skip_window`
`45`	`45`	`extend_words.extend(words)`
`46`	`46`	`extend_words.extend([2] * self.skip_window)`

Comments

(0)