Skip to content

Commit

Permalink
minor change
Browse files Browse the repository at this point in the history
  • Loading branch information
kazutotess committed Sep 9, 2020
1 parent b4d16ca commit 226add9
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 10 deletions.
10 changes: 3 additions & 7 deletions LSTM_with_shape.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,7 @@ def LSTM_with_shape(act, data_size, layer_num, unit_num, return_sequence,

def main():
# specify GPU
# you need to coment out this part if you don't use GPU
config = tf.ConfigProto(
gpu_options=tf.GPUOptions(
allow_growth=True,
Expand Down Expand Up @@ -123,15 +124,10 @@ def main():
assert num_of_ts + time_step * (maxlen - 1) < \
num_of_ts_for_data, 'The data aumont is not enough.'

data_LSTM = pd.read_csv(
path_data, header=None, delim_whitespace=False
)
data_LSTM = pd.read_csv(path_data, header=None, delim_whitespace=False)
data_LSTM = data_LSTM.values

X_CNN = np.zeros([
number_of_shape * num_of_ts,
120, 120, 1
])
X_CNN = np.zeros([number_of_shape * num_of_ts, 120, 120, 1])
for i in range(number_of_shape):
data_CNN = pd.read_csv(
path_to_present_dir +
Expand Down
2 changes: 1 addition & 1 deletion Muliti-Scale_CNN-AE.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def main():
x_num, y_num, phys_num])

for i in tqdm(range(1, kind_num + 1)):
fnstr = path_to_present_dir + '/data/pickles/data_' + \
fnstr = path_to_present_dir + '/data/CNNAE/data_' + \
'{0:03d}'.format(i)+'.pickle'

# Pickle load
Expand Down
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,8 @@ The code is written for educational clarity and not for speed.

# Directory structure
ML-ROM_Various_Shapes ── CNN_autoencoder/
├─ data ─── pickles ─── data_001.pikle ~ data080.pickle
│ │ └─ Test_data/data_001.pikle ~ data020.pickle
├─ data ─── CNNAE ─── data_001.pikle ~ data080.pickle
│ │ └─ Test_data/data_001.pikle ~ data020.pickle
│ └─ LSTM ─── Dataset/
│ └─ Flags/
├─ .gitignore
Expand Down

0 comments on commit 226add9

Please sign in to comment.