Esempio n. 1
0
num_of_CUEs = 2
num_of_D2Ds = 2
batch_size = 64
epochs = 10

# Get the image data format which Keras follows
image_data_format = Preprocessing.GetImageDataFormat()

# Get the input data and target data
input_data = Preprocessing.GetInputData(num_of_cells, num_of_CUEs, num_of_D2Ds,
                                        (2000, 8000, 10000), image_data_format)
target_data = Preprocessing.GetTargetData(num_of_cells, num_of_CUEs,
                                          num_of_D2Ds, (2000, 8000, 10000))

# Reshape the input data
reshaped_input_data = Preprocessing.ReshapeInputData1D(input_data)

# Split the datadset into the training set and testing set
(x_train, y_train), (x_test,
                     y_test) = Preprocessing.SplitDataset(reshaped_input_data,
                                                          target_data,
                                                          proportion=0.8,
                                                          shuffle=False)

# Get the input shape of input data and the output shape of target data
input_shape = Preprocessing.GetInputShape(x_train)
target_shape = Preprocessing.GetTargetShape(y_train)

# Build the model
model = Sequential()