Skip to content

Instantly share code, notes, and snippets.

@amansrivastava17

amansrivastava17/test_model.py

Created November 12, 2019 07:16
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save amansrivastava17/e1f25b1d7ac74ef9c0040931ed1e7bbb to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save amansrivastava17/e1f25b1d7ac74ef9c0040931ed1e7bbb to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. amansrivastava17 created this gist Nov 12, 2019.
    11 changes: 11 additions & 0 deletions test_model.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,11 @@
    def test(sentence, model_path, word_index_path)
    classifier = models.load_model( 'models/models.h5' )
    tokenizer = tf.keras.preprocessing.text.Tokenizer(filters='.,:?{} ')
    sentences = re.sub(r'.,:?{}', ' ', sentence)

    with open(word_index_path, 'r') as f:
    tokenizer.word_index = json.loads(f.read())
    tokenized_messages = tokenizer.texts_to_matrix(sentence.split())
    p = list(classifier.predict(tokenized_messages)[0])
    for index, each in enumerate(p):
    print(index, each)