-
-
Notifications
You must be signed in to change notification settings - Fork 19
2nd phase Evaluations #19
base: develop
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I've pointed out some notable changes.
@hardlyhuman , please review the changes
|
||
return model, score | ||
|
||
def save_model(self,model,filename): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This one is for saving the models
iids/classifier/nnclassifier.py
Outdated
from torch.autograd import Variable | ||
|
||
|
||
class CNN(nn.Module): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
added all classes of neural nets in a single files
iids/classifier/nnclassifier.py
Outdated
bmu_loc = bmu_loc.squeeze() | ||
|
||
|
||
# Neural network parameters " would be collected from JSON Config" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I Would be making the respective changes in the views.py
iids/classifier/nnclassifier.py
Outdated
#model = RNN(1, 23) | ||
#model = AE(1, 23) | ||
|
||
def neural_train(): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This function is for training of only neural networks
|
||
#for Neural Networks | ||
|
||
#"""Convert the string part of the data into numbers, and convert the input 41-dimensional features into an 8*8 matrix """ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
added changes for neural nets
This PR contains only the changes made in 2nd phase.
Added a classifer for neural nets.
Updated mlclassifiers by addition of ML algos.
Updated pre-processors for neural nets.