diff --git a/skluc/main/tensorflow_/utils.py b/skluc/main/tensorflow_/utils.py index c184d2760fd65deff2a8eaf36255b56ad4c021c5..9dffd68a44fa5ef39b9db6f37cc6508e2a38693c 100644 --- a/skluc/main/tensorflow_/utils.py +++ b/skluc/main/tensorflow_/utils.py @@ -248,8 +248,6 @@ def replace_nan(tensor): return tf.where(tf.is_nan(tensor), tf.zeros_like(tensor), tensor) - - if __name__ == "__main__": X_train = np.arange(12) Y_train = np.array(["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p"])