Tensorflow-Keras reproducibility problem on Google Colab

前端 未结 2 661
佛祖请我去吃肉
佛祖请我去吃肉 2021-01-22 16:13

I have a simple code to run on Google Colab (I use CPU mode):

import numpy as np
import pandas as pd

## LOAD DATASET

datatrain = pd.read_csv(\"gdrive/My Drive/         


        
2条回答
  •  梦谈多话
    2021-01-22 16:40

    You should also fix the seed for kernel_initializer in your Dense layers. So, your model will be like:

    model = tf.keras.Sequential()
    model.add(Dense(10, kernel_initializer=keras.initializers.glorot_uniform(seed=66), input_shape=(4,)))
    model.add(Activation("sigmoid"))
    model.add(Dense(3, kernel_initializer=keras.initializers.glorot_uniform(seed=66)))
    model.add(Activation("softmax"))
    

提交回复
热议问题