Commit a0fb9322 authored by Jun Zhao's avatar Jun Zhao
Browse files

version control

parent 47c6c7f6
Loading
Loading
Loading
Loading
+5 −2
Original line number Diff line number Diff line
import numpy as np
import tensorflow as tfcheck
if tfcheck.__version__[0] == '2':
  import tensorflow.compat.v1 as tf
else:
  import tensorflow as tf
# import optuna
import os
from sklearn.model_selection import train_test_split