美文网首页
Tensorflow - Keras Demo

Tensorflow - Keras Demo

作者: YANWeichuan | 来源:发表于2018-08-03 18:35 被阅读0次

    第一次运行 tensorflow的keras demo,发现loss不减少,精度不提高

    $ python basic-classification.py
    1.8.0
    (60000, 28, 28)
    Epoch 1/5
    2018-07-24 13:36:01.051139: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
    60000/60000 [==============================] - 3s 52us/step - loss: 14.5062 - acc: 0.1000
    Epoch 2/5
    60000/60000 [==============================] - 3s 51us/step - loss: 14.5063 - acc: 0.1000
    Epoch 3/5
    60000/60000 [==============================] - 3s 52us/step - loss: 14.5063 - acc: 0.1000
    Epoch 4/5
    60000/60000 [==============================] - 3s 51us/step - loss: 14.5063 - acc: 0.1000
    Epoch 5/5
    60000/60000 [==============================] - 3s 52us/step - loss: 14.5063 - acc: 0.1000
    10000/10000 [==============================] - 0s 29us/step
    

    修改激励函数relu 为sigmoid

    model = keras.Sequential([   
    
        keras.layers.Flatten(input_shape=(28, 28)),   
    
        keras.layers.Dense(128, activation=tf.nn.sigmoid),   
    
        keras.layers.Dense(10, activation=tf.nn.softmax)
    
    ])
    

    原因:未明!

    相关文章

      网友评论

          本文标题:Tensorflow - Keras Demo

          本文链接:https://www.haomeiwen.com/subject/huyuvftx.html