s487194/ium_487194
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

mlflow fix2

Browse Source
This commit is contained in:
Witold Woch 2023-05-14 23:23:58 +02:00
parent e394fdca51
commit 5c28e58e2b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -41,7 +41,7 @@ X_test = torch.FloatTensor(X_test)
y_train = torch.LongTensor(y_train) y_train = torch.LongTensor(y_train)
y_test = torch.LongTensor(y_test) y_test = torch.LongTensor(y_test)
#### Model ### Model
class ANN_Model(nn.Module): class ANN_Model(nn.Module):
def __init__(self,input_features=82,hidden1=20,hidden2=20,out_features=3): def __init__(self,input_features=82,hidden1=20,hidden2=20,out_features=3):