Skip to content

1 Dimensional Convolutional Neural Network for Iris dataset classification

License

Notifications You must be signed in to change notification settings

cserajdeep/1DCNN-IRIS-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

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

1dcnniris_auc

1DCNN IRIS PyTorch

Releases

No releases published

Packages

No packages published