search

GoogleCloudPlatform / ml-design-patterns

Source code accompanying O'Reilly book: Machine Learning Design Patterns
Apache License 2.0
1.87k stars 527 forks source link

Chapter 5: Continued Evaluation - Class label consistency #12

Closed mshearer0 closed 4 years ago

mshearer0 commented 4 years ago

The order of the publications is inconsistent between the original CLASSES definition and the source name function:

CLASSES = { 'github': 0, 'nytimes': 1, 'techcrunch': 2 }

labels = tf.constant(['github', 'techcrunch', 'nytimes'], dtype=tf.string)

Suggest:

labels = tf.constant(['github', 'nytimes','techcrunch'], dtype=tf.string)

munnm commented 4 years ago

Thanks for catching this. I've fixed it in the notebook.