Datasetpublic

ysn-rfd/text-dataset-tiny-code-script-py-format

USED of tahamajs/medicine_ds_persian for .parquet file USED of Alijafarixcs2/persian-it-llama2-2k for .parquet file USED of Abirate/english_quotes for .jsonl file NEW FILES (05/12/2025) NEW FILES (12/26/2025) NEW FILES (02/15/2026)

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face

3likes1.7kdownloads

rootpytorch_studyday5_1.py

day5_1.py28 linesDownload Raw Back to pytorch_study

1import torch
2import torch.nn as nn
3import torch.nn.functional as F
4import torch.optim as optim
5
6class neural_network (nn.Module):
7    def __init__(self, input_dim, hidden_dim, output_dim):
8        super(neural_network, self).__init__()
9        self.hidden = nn.Linear (input_dim, hidden_dim)
10        self.act = nn.ReLU()
11        self.output = nn.Linear (hidden_dim, output_dim)
12
13    def forward (self, x):
14        x = self.hidden (x)
15        x = self.act (x)
16        x = self.output (x)
17        return x
18
19
20input_dim = 4
21hidden_dim = 32
22output_dim = 4
23
24model = neural_network(input_dim, hidden_dim, output_dim)
25print(model)
26
27criterion = nn.CrossEntropyLoss()
28optimizer = optim.Adam(model.parameters(), lr= 0.01)