ballet.validation.feature_pruning.validator module¶

class ballet.validation.feature_pruning.validator.GFSSFPruner(*args, lmbda_1=0.0, lmbda_2=0.0, lambda_1_adjustment=64, lambda_2_adjustment=64)[source]¶

Bases: ballet.validation.base.FeaturePruningMixin, ballet.validation.gfssf.GFSSFPerformanceEvaluator

prune()[source]¶

Prune using GFSSF

Uses lines 12-13 of agGFSSF

class ballet.validation.feature_pruning.validator.NoOpPruner(X_df, y_df, X_df_val, y_val, features, candidate_feature)[source]¶

Bases: ballet.validation.base.FeaturePruner

prune()[source]¶: Prune existing features, returning list of features to remove

class ballet.validation.feature_pruning.validator.RandomPruner(*args, p=0.3, seed=None)[source]¶

Bases: ballet.validation.base.FeaturePruningMixin, ballet.validation.common.RandomFeaturePerformanceEvaluator

prune()[source]¶: With probability p, select a random feature to prune