search

sigp / blockprint

Block fingerprinting for the beacon chain, for client diversity metrics
Apache License 2.0
91 stars 15 forks source link

Support mlp classifier #33

Closed santi1234567 closed 5 months ago

santi1234567 commented 5 months ago

Description

Added support for utilizing an Multi-Layer Perceptron classifier (MLP). The knn_classifier module is now called classifier and supports both algorithms with the optional classifier_type parameter on the class. There is also the new --classifier-type flag in the classifier.py script.

The MLP classifier has proven to achieve at least 3% more accuracy over the KNN one with most configurations tested by us. The only downside is that it takes more time to be trained.

michaelsproul commented 5 months ago

Looks good in terms of structure and everything @santi1234567! Once CI is passing I'm happy to merge