bugfinder.features.reduction.auto_encoder

Module for the auto-encoder

class bugfinder.features.reduction.auto_encoder.FeatureSelector(dataset)

Bases: AbstractFeatureSelector

Feature selection with auto-encoder

select_feature(input_features, input_results, dry_run, dimension, layers, encoder_path) → DataFrame

Feature selection algorithm

train_encoder(input_features, dimension, hidden_layers)

Training function for the auto-encoder.