MicEMD.classification package¶
Submodules¶
MicEMD.classification.ANN module¶
The ANN classification method in TDEM
- Methods:
MLP: the ANN classification method
- MicEMD.classification.ANN.MLP(train_set, test_set, solver='lbfgs', hidden_layer_sizes=(50), activation='tanh')¶
the ANN classification algorithm
- train_set: ndarry
the train set
- test_set: ndarry
the test set
- solver: str
the optimization of the ANN
- hidden_layer_sizes: tuple
the ANN hidden layer sizes
- activation: str
the activation of the ANN
- res: tuple
(accuracy, y_pred, y_true),the accuracy of classification, the predict value and the true value