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

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -11
requirements.txt CHANGED
@@ -1,28 +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.23.0 # Numerical operations (required for model input/output processing)
4
- pandas==1.5.2 # Data manipulation and preprocessing
5
 
6
  # Plotting and Visualization Tools
7
- matplotlib==3.7.0 # Visualization library (used for plotting confusion matrices)
8
- seaborn==0.12.1 # Advanced data visualization, helpful for heatmaps (confusion matrix)
9
 
10
  # Saving and Loading Models
11
- joblib==1.2.0 # For saving and loading machine learning models (used for Random Forest, Decision Trees, etc.)
12
-
13
- # Reporting and Metrics
14
- scikit-learn==1.5.1 # For generating classification reports, confusion matrices, and model evaluation metrics
15
 
16
  # Hugging Face Hub Integration
17
  huggingface_hub==0.29.0rc7 # Integration with Hugging Face Hub (for model uploading, downloading, sharing)
18
  transformers==4.26.1 # Hugging Face Transformers library (for model usage on the Hub)
19
 
20
  # Optional - Jupyter Notebooks for Model Development and Experimentation
21
- notebook==7.0.0 # For running Jupyter Notebooks in your project
22
 
23
  # Optional - TensorBoard for Visualizing Training Process (if applicable to larger models)
24
  tensorboard==2.10.1 # For tracking and visualizing model training
25
 
26
  # Extras for performance and speedups
27
- xgboost==1.6.2 # Gradient boosting library (optional, if you want to use advanced tree-based models)
28
- lightgbm==3.3.5 # LightGBM for fast gradient boosting (optional, for high performance)
 
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)