|
12 | 12 | print (parameterString)
|
13 | 13 | print ("filename is ", filename)
|
14 | 14 |
|
15 |
| -epochs = int(parameterString[0].strip()) |
16 |
| -batch_size = int(parameterString[2].strip()) |
17 |
| -print ("Running p1b1 for epochs ", epochs, batch_size) |
18 | 15 |
|
19 |
| -# N1 = int(parameterString[2].strip()) |
20 |
| -# NE = int(parameterString[3].strip()) |
| 16 | +integs = [int(x) for x in parameterString.split(',')] |
| 17 | +print (integs) |
21 | 18 |
|
22 |
| -print("Set the correct paths for test and train file") |
23 |
| -test_path="/home/jain/Benchmarks/Data/Pilot1/P1B1.test.csv" |
24 |
| -train_path="/home/jain/Benchmarks/Data/Pilot1/P1B1.train.csv" |
| 19 | +epochs = integs[0] |
| 20 | +batch_size = integs[1] |
| 21 | +N1 = integs[2] |
| 22 | +NE = integs[3] |
25 | 23 |
|
26 | 24 | print ("Starting to loading Xtrain and Xtest")
|
27 |
| -X_train, X_test = p1b1.load_data(test_path=test_path, train_path=train_path) |
| 25 | +X_train, X_test = p1b1.load_data() |
28 | 26 | print ("Done loading Xtrain and Xtest")
|
29 | 27 |
|
30 |
| -print ("Running p1b1 for epochs ", epochs) |
| 28 | +print ("Running p1b1 for epochs, batch_size, N1, NE", epochs, batch_size, N1, NE) |
| 29 | +# Need to introduce N1 and NE as parameters to the run_p1b1 function |
31 | 30 | encoder, decoder, history = p1b1_baseline_keras2.run_p1b1(X_train, X_test, epochs=epochs, batch_size=batch_size)
|
32 |
| -print ("Done running p1b1 for epochs ", epochs) |
| 31 | +print ("Done running p1b1") |
33 | 32 |
|
34 | 33 | # works around this error:
|
35 | 34 | # https://github.com/tensorflow/tensorflow/issues/3388
|
|
0 commit comments