main.py

import numpy as np
import tensorflow as tf
import pickle as pickle # python pkl 文件读写

from CNNs_model import cnn_model_fn

tf.logging.set_verbosity(tf.logging.INFO)


def main(unused_argv):


    predict_data = np.array(pickle.load(open('cache/test_temporary_data.plk', 'rb')) )
    predict_labels = np.array(pickle.load(open('cache/test_temporary_labels.plk', 'rb')) )
    
    print(predict_data.shape)

    # predict_data = np.array(pickle.load(open('cache/test_data.plk', 'rb')) )
    # predict_labels = np.array(pickle.load(open('cache/test_labels.plk', 'rb')) )

    # with tf.Session() as sess:
    #     train_data = tf.convert_to_tensor(train_data_np)
    #     eval_data = tf.convert_to_tensor(eval_data_np)

    # Create the Estimator
    cnn_classifier = tf.estimator.Estimator(
        # model_fn=cnn_model_fn, model_dir="cnn_convnet_model")
        model_fn=cnn_model_fn, model_dir="Model/cnn")


    # # Evaluate the model and print results
    # eval_input_fn = tf.estimator.inputs.numpy_input_fn(
    #     x={"x": predict_data},
    #     y=predict_labels,
    #     num_epochs=1,
    #     shuffle=False) 
    # eval_results = cnn_classifier.evaluate(input_fn=eval_input_fn)
    # print(eval_results)

    # predict = np.empty([1,30000])
    # predict[0] = np.array(predict_data[0])
    # print(predict.shape)
    # predict
    predict_input_fn = tf.estimator.inputs.numpy_input_fn(
        x={"x": predict_data},
        num_epochs=1,
        shuffle=False) 
    predict_results = cnn_classifier.predict(input_fn=predict_input_fn)


    tag_predict = [] 
    softmax = []
    probability = []
    # print(type(predict_results))
    for e in  predict_results:
        # print(e['classes'])
        # print(e['probabilities'])
        tag_predict.append(e['classes'])
        softmax.append(e['probabilities'])
        probability_one = 0
        for i in  e['probabilities']:
            if probability_one < i : probability_one = i
        probability.append(probability_one)
    # print("done!")
    # for i in predict_labels:
        # print(i)

    No = 5
    print("tag:", tag_predict[No])
    print("tag_predict:", tag_predict[No])
    print("probability:", probability[No])
    print("softmax", softmax[No])

if __name__ == "__main__":
    
    tf.app.run()