Zaktualizuj 'DL.py'
This commit is contained in:
parent
434c69044c
commit
29942de8df
2
DL.py
2
DL.py
@ -8,8 +8,6 @@ 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):
|
||||||
|
Loading…
Reference in New Issue
Block a user