An open API service indexing awesome lists of open source software.

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

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
1dcnniris_auc

# 1DCNN IRIS PyTorch