cserajdeep / 1DCNN-IRIS-PyTorch

1 Dimensional Convolutional Neural Network for Iris dataset classification

1D-CNN

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