Zaktualizuj 'DL.py'

This commit is contained in:
Kornelia Girejko 2022-05-06 20:09:09 +02:00
parent fa881a8767
commit 087df12aa7

2
DL.py
View File

@ -8,6 +8,8 @@ from sklearn import preprocessing
from sklearn.preprocessing import StandardScaler from sklearn.preprocessing import StandardScaler
scaler = StandardScaler() scaler = StandardScaler()
EPOCHS = int(sys.argv[1])
# Model # Model
class Model(nn.Module): class Model(nn.Module):
def __init__(self): def __init__(self):