Skip to content

This project provides a comprehensive toolset for feature selection using LightGBM, a gradient boosting framework that uses tree-based learning algorithms. The primary goal is to improve model performance by selecting the most relevant features and discarding the redundant ones.

License

Notifications You must be signed in to change notification settings

YigitTurali/AFS_BM-Algorithm

Repository files navigation

AFS_BM-Algorithm Repository

This project provides a comprehensive toolset for feature selection using LightGBM, a gradient-boosting framework that uses tree-based learning algorithms. The primary goal is to improve model performance by selecting the most relevant features and discarding the redundant ones.

Directories and Their Descriptions:

1. DataLoaders

Description: This directory contains custom data loaders for classification and time series data.

  • classification_dataloader.py: For loading classification datasets.
  • time_series_dataloader.py: For loading time series datasets.
  • m4_feature_extractor.py: Extracts features from the M4 dataset.

2. Models

Description: This directory houses implementations of different machine-learning models and pipelines.

  • Feature_Selector_LightGBM.py: Feature selection using LightGBM.
  • Feature_Selector_MLP.py: Feature selection using MLP.
  • LightGBM_Pipeline.py: A pipeline for LightGBM.
  • MLP_Pipeline.py: A pipeline for MLP.
  • XGBoost_Pipeline.py: A pipeline for XGBoost.

3. Results

Description: (The content didn't provide specific details about this directory. You might want to describe what kind of results or output files are stored here.)

4. Runners

Description: This directory contains scripts to run the feature selector models.

  • LightGBM_Feature_Selector_Runner.py: Runner for the LightGBM feature selector.
  • MLP_Feature_Selector_Runner.py: Runner for the MLP feature selector.

Installation

  1. Clone the repository: git clone https://github.com/YigitTurali/AFS_BM-Algorithm.git

  2. Navigate to the repository directory: cd AFS_BM-Algorithm

  3. Install the required packages: pip install -r requirements.txt

Usage

  1. Load your dataset using the appropriate data loader from the DataLoaders directory.
  2. Choose the model or pipeline you want to use from the Models directory.
  3. Run the corresponding runner script from the root directory to start the feature selection process.

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

License

This project is licensed under the MIT License.

About

This project provides a comprehensive toolset for feature selection using LightGBM, a gradient boosting framework that uses tree-based learning algorithms. The primary goal is to improve model performance by selecting the most relevant features and discarding the redundant ones.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages