s444356/ium_444356
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update dllib-mlflow.py
All checks were successful
s444356-evaluation/pipeline/head This commit looks good
Details
s444356-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Maciej Czajka 2022-05-11 19:06:07 +02:00
parent 26b0736758
commit b865d30572
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Biblioteka_DL/dllib-mlflow.py
View File

@ -31,10 +31,10 @@ class Model(nn.Module):
self.layer3 = nn.Linear(40, 3) self.layer3 = nn.Linear(40, 3)
def forward(self, x): def forward(self, x):
x = F.relu(self.layer1(x)) x = F.relu(self.layer1(x.float()))
x = F.relu(self.layer2(x)) x = F.relu(self.layer2(x.float()))
x = F.softmax(self.layer3(x)) # To check with the loss function x = F.softmax(self.layer3(x.float())) # To check with the loss function
return x return x.float()
# funkcja usuwająca wiersze zawierające platformę "Stadia" # funkcja usuwająca wiersze zawierające platformę "Stadia"
def delete_stadia(games): def delete_stadia(games):