Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def split_input_target(chunk): # Define a function to split the input and target
|
|
26 |
dataset = sequences.map(split_input_target) # Apply the function to the dataset
|
27 |
|
28 |
# Shuffle and batch the dataset
|
29 |
-
BATCH_SIZE =
|
30 |
BUFFER_SIZE = 10000 # Buffer size for shuffling
|
31 |
dataset = dataset.shuffle(BUFFER_SIZE).batch(BATCH_SIZE, drop_remainder=True) # Shuffle and batch the dataset
|
32 |
|
|
|
26 |
dataset = sequences.map(split_input_target) # Apply the function to the dataset
|
27 |
|
28 |
# Shuffle and batch the dataset
|
29 |
+
BATCH_SIZE = 1 # Batch size
|
30 |
BUFFER_SIZE = 10000 # Buffer size for shuffling
|
31 |
dataset = dataset.shuffle(BUFFER_SIZE).batch(BATCH_SIZE, drop_remainder=True) # Shuffle and batch the dataset
|
32 |
|