This repository was archived by the owner on Mar 10, 2023. It is now read-only.

Commit 6e78c26

committed

Added sequence_decoder function.

1 parent 8cd24de commit 6e78c26Copy full SHA for 6e78c26

File tree

1 file changed

+21

-5

lines changed

train.py

1 file changed

+21

-5

lines changed

`‎train.py‎`

Lines changed: 21 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -208,6 +208,26 @@ def read_text_files(dir, extensions=['txt']):`
`208`	`208`	`return files`
`209`	`209`
`210`	`210`
	`211`	`+def sequence_decoder(sequence, first_index=(ord('a') - 1)):`
	`212`	`+ """`
	`213`	`+ Read text files.`
	`214`	`+`
	`215`	`+ Args:`
	`216`	`+ sequence: list of int.`
	`217`	`+ Encoded sequence`
	`218`	`+ first_index: int.`
	`219`	`+ First index (usually index of 'a').`
	`220`	`+ Returns:`
	`221`	`+ decoded_text: string.`
	`222`	`+ """`
	`223`	`+ decoded_text = ''.join([chr(x) for x in np.asarray(sequence) + first_index])`
	`224`	`+ # Replacing blank label to none.`
	`225`	`+ decoded_text = decoded_text.replace(chr(ord('z') + 1), '')`
	`226`	`+ # Replacing space label to space.`
	`227`	`+ decoded_text = decoded_text.replace(chr(ord('a') - 1), ' ')`
	`228`	`+ return decoded_text`
	`229`	`+`
	`230`	`+`
`211`	`231`	`def main(argv):`
`212`	`232`	`# Read text file.`
`213`	`233`	`text_file_path = TRAIN_DIR + "211-122425-0059.txt"`
`@@ -331,11 +351,7 @@ def main(argv):`
`331`	`351`
`332`	`352`	`# Decoding.`
`333`	`353`	`decoded_outputs = session.run(decoded[0], feed_dict=feed)`
`334`		`- decoded_text = ''.join([chr(x) for x in np.asarray(decoded_outputs[1]) + FIRST_INDEX])`
`335`		`- # Replacing blank label to none.`
`336`		`- decoded_text = decoded_text.replace(chr(ord('z') + 1), '')`
`337`		`- # Replacing space label to space.`
`338`		`- decoded_text = decoded_text.replace(chr(ord('a') - 1), ' ')`
	`354`	`+ decoded_text = sequence_decoder(decoded_outputs[1])`
`339`	`355`
`340`	`356`	`logging.info("Original:\n%s", text)`
`341`	`357`	`logging.info("Decoded:\n%s", decoded_text)`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 6e78c26

File tree

1 file changed

1 file changed

`‎train.py‎`

0 commit comments