SilverDragon9 commited on
Commit
a69aa6d
·
verified ·
1 Parent(s): ece8ef7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -7
requirements.txt CHANGED
@@ -1,25 +1,25 @@
1
  # Core Libraries for Machine Learning
2
- scikit-learn==1.5.1 # Essential library for machine learning models (Random Forest, Decision Trees, etc.)
3
  numpy==1.26.4 # Numerical operations (required for model input/output processing)
4
- pandas==2.2.2 # Data manipulation and preprocessing
5
 
6
  # Plotting and Visualization Tools
7
- matplotlib==3.9.2 # Visualization library (used for plotting confusion matrices)
8
  seaborn==0.13.2 # Advanced data visualization, helpful for heatmaps (confusion matrix)
9
 
10
  # Saving and Loading Models
11
  joblib==1.4.2 # For saving and loading machine learning models (used for Random Forest, Decision Trees, etc.)
12
 
13
  # Hugging Face Hub Integration
14
- huggingface_hub==0.29.0rc7 # Integration with Hugging Face Hub (for model uploading, downloading, sharing)
15
  transformers==4.26.1 # Hugging Face Transformers library (for model usage on the Hub)
16
 
17
  # Optional - Jupyter Notebooks for Model Development and Experimentation
18
- notebook==7.2.2 # For running Jupyter Notebooks in your project
19
 
20
  # Optional - TensorBoard for Visualizing Training Process (if applicable to larger models)
21
- tensorboard==2.10.1 # For tracking and visualizing model training
22
 
23
  # Extras for performance and speedups
24
- xgboost==2.1.4 # Gradient boosting library (optional, if you want to use advanced tree-based models)
25
  lightgbm==4.6.0 # LightGBM for fast gradient boosting (optional, for high performance)
 
1
  # Core Libraries for Machine Learning
2
+ scikit-learn==1.6.1 # Essential library for machine learning models (Random Forest, Decision Trees, etc.)
3
  numpy==1.26.4 # Numerical operations (required for model input/output processing)
4
+ pandas==2.2.3 # Data manipulation and preprocessing
5
 
6
  # Plotting and Visualization Tools
7
+ matplotlib==3.10.1 # Visualization library (used for plotting confusion matrices)
8
  seaborn==0.13.2 # Advanced data visualization, helpful for heatmaps (confusion matrix)
9
 
10
  # Saving and Loading Models
11
  joblib==1.4.2 # For saving and loading machine learning models (used for Random Forest, Decision Trees, etc.)
12
 
13
  # Hugging Face Hub Integration
14
+ huggingface_hub==0.30.2 # Integration with Hugging Face Hub (for model uploading, downloading, sharing)
15
  transformers==4.26.1 # Hugging Face Transformers library (for model usage on the Hub)
16
 
17
  # Optional - Jupyter Notebooks for Model Development and Experimentation
18
+ notebook==7.2.2 # For running Jupyter Notebooks in your project
19
 
20
  # Optional - TensorBoard for Visualizing Training Process (if applicable to larger models)
21
+ tensorboard==2.19.1 # For tracking and visualizing model training
22
 
23
  # Extras for performance and speedups
24
+ xgboost==3.0.0 # Gradient boosting library (optional, if you want to use advanced tree-based models)
25
  lightgbm==4.6.0 # LightGBM for fast gradient boosting (optional, for high performance)