Update README.md
Browse files
README.md
CHANGED
|
@@ -16,6 +16,7 @@ This model is extremely weak. I am not good at data science
|
|
| 16 |
|
| 17 |
<details>
|
| 18 |
<summary><b>Trained on 500 Epoch with 2.1 million song data from Spotify Database</b></summary>
|
|
|
|
| 19 |
```
|
| 20 |
import torch
|
| 21 |
import torch.nn as nn
|
|
@@ -86,6 +87,7 @@ for epoch in range(num_epochs):
|
|
| 86 |
model.eval()
|
| 87 |
with torch.no_grad():
|
| 88 |
predicted = model(X_test_tensor)
|
|
|
|
| 89 |
```
|
| 90 |
|
| 91 |
</details>
|
|
|
|
| 16 |
|
| 17 |
<details>
|
| 18 |
<summary><b>Trained on 500 Epoch with 2.1 million song data from Spotify Database</b></summary>
|
| 19 |
+
|
| 20 |
```
|
| 21 |
import torch
|
| 22 |
import torch.nn as nn
|
|
|
|
| 87 |
model.eval()
|
| 88 |
with torch.no_grad():
|
| 89 |
predicted = model(X_test_tensor)
|
| 90 |
+
|
| 91 |
```
|
| 92 |
|
| 93 |
</details>
|