fix

2022-05-02 22:56:24 +02:00 · 2022-05-02 22:56:24 +02:00 · 81ef5f8c7d
commit 81ef5f8c7d
parent 3f7a6d212f
1 changed files with 4 additions and 0 deletions
--- a/Scripts/train_neural_network.py
+++ b/Scripts/train_neural_network.py
@ -86,6 +86,10 @@ def load_data(data_path, filename) -> pd.DataFrame:
 def read_params():
    global data_path, num_words, epochs, batch_size, pad_length
    data_path, num_words, epochs, batch_size, pad_length = sys.argv[1].split(',')
+    num_words = int(num_words)
+    epochs = int(epochs)
+    batch_size = int(batch_size)
+    pad_length = int(pad_length)


 def main():