s444439/ium_z444439
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add parameter fix

Browse Source
This commit is contained in:
Klaudia 2023-05-11 19:02:39 +02:00
parent 303bfb0193
commit 21fc9d3677
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
create-dataset.py
View File

@ -23,4 +23,4 @@ Y_test.to_csv('Y_test.csv', index=False)
Y_train.to_csv('Y_train.csv', index=False) Y_train.to_csv('Y_train.csv', index=False)
Y_dev.to_csv('Y_dev.csv', index=False) Y_dev.to_csv('Y_dev.csv', index=False)
train.main() train.main(0)

8
train.py
View File

@ -5,8 +5,9 @@ import tensorflow
from keras.applications.densenet import layers from keras.applications.densenet import layers
def main(): def main(EPOCHS):
EPOCHS = int(os.environ['EPOCHS']) if EPOCHS == 0:
EPOCHS = 500
train_data_x = pd.read_csv('./X_train.csv') train_data_x = pd.read_csv('./X_train.csv')
adults_train = train_data_x.copy() adults_train = train_data_x.copy()
@ -30,4 +31,5 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
main() EPOCHS = int(os.environ['EPOCHS'])
main(EPOCHS)