# Set the keeping probability
keep_prob = 0.7