https://github.com/cserajdeep/1dcnn-iris-pytorch
1 Dimensional Convolutional Neural Network for Iris dataset classification
https://github.com/cserajdeep/1dcnn-iris-pytorch
1d-cnn classification cnn-classification deep-neural-networks iris-dataset python pytorch
Last synced: 30 days ago
JSON representation
1 Dimensional Convolutional Neural Network for Iris dataset classification
- Host: GitHub
- URL: https://github.com/cserajdeep/1dcnn-iris-pytorch
- Owner: cserajdeep
- License: mit
- Created: 2021-07-08T03:54:58.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2021-07-08T03:58:28.000Z (almost 4 years ago)
- Last Synced: 2025-03-27T22:39:56.461Z (about 2 months ago)
- Topics: 1d-cnn, classification, cnn-classification, deep-neural-networks, iris-dataset, python, pytorch
- Language: Jupyter Notebook
- Homepage:
- Size: 99.6 KB
- Stars: 13
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 1D-CNN
```ruby
class OneDCNN(nn.Module):
def __init__(self, num_classes, in_channels, out_channels, kernel_size=3, stride=1):
super(OneDCNN,self).__init__()
self.in_channels = in_channels
self.out_channels = out_channels
self.kernel_size = kernel_size
self.conv1d = nn.Conv1d(in_channels, out_channels, kernel_size)
self.relu = nn.ReLU(inplace=True)
self.fc1 = nn.Linear(out_channels,50)
self.fc2 = nn.Linear(50,num_classes)
```# Area Under Curve
# 1DCNN IRIS PyTorch