https://github.com/cserajdeep/lstm-iris-pytorch

LSTM implementation on IRIS dataset classification using PyTorch
https://github.com/cserajdeep/lstm-iris-pytorch

classification deep-neural-networks iris-dataset lstm lstm-neural-networks python pytorch

Last synced: 3 months ago
JSON representation

LSTM implementation on IRIS dataset classification using PyTorch

• Host: GitHub
• URL: https://github.com/cserajdeep/lstm-iris-pytorch
• Owner: cserajdeep
• License: mit
• Created: 2021-07-08T04:00:56.000Z (about 5 years ago)
• Default Branch: main
• Last Pushed: 2021-07-08T04:03:07.000Z (about 5 years ago)
• Last Synced: 2025-03-27T22:39:52.387Z (over 1 year ago)
• Topics: classification, deep-neural-networks, iris-dataset, lstm, lstm-neural-networks, python, pytorch
• Language: Jupyter Notebook
• Homepage:
• Size: 106 KB
• Stars: 1
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# LSTM Model

```ruby

class LSTM(nn.Module):

  def __init__(self, num_classes, input_size, hidden_size, num_layers):

    super(LSTM, self).__init__()

    self.num_classes = num_classes

    self.num_layers = num_layers

    self.input_size = input_size

    self.hidden_size = hidden_size

    self.seq_length = seq_length

        

    self.lstm = nn.LSTM(input_size=input_size, hidden_size=hidden_size,

                            num_layers=num_layers, batch_first=True)

        

    self.fc = nn.Linear(hidden_size, num_classes)

```

# Area Under Curve



# LSTM IRIS PyTorch