Downloads · 30 days
0
vikashHugFace/engine-failure-prediction-model
engine-failure-prediction-model is a machine learning model from vikashHugFace. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
--- tags: - classification - engine-failure-prediction - sklearn metrics: - accuracy - precision - recall - f1 - rocauc ---
Downloads · 30 days
0
Access
Public
Updated Jul 5, 2026
Repo size
9.1 KB
Likes
0
Public
Click a slice to open those files.
.pkl4.5 KB · 60%
From the Hugging Face model README
tags:
- classification
- engine-failure-prediction
- sklearn
metrics:
- accuracy
- precision
- recall
- f1
- roc_auc
---
# Engine Failure Prediction Model
**Algorithm:** Decision Tree
**Task:** Binary Classification — predict `Engine Condition`
## Performance (held-out test set)
| Metric | Value |
|-----------|--------|
| Accuracy | 0.6404 |
| Precision | 0.6871 |
| Recall | 0.7889 |
| F1-Score | 0.7345 |
| ROC AUC | 0.6687 |
## Best Hyperparameters
```json
{
"model__max_depth": 3, "model__min_samples_leaf": 1, "model__min_samples_split": 2 } ```
## Usage
```python
import joblib
from huggingface_hub import hf_hub_download
path = hf_hub_download(repo_id="vikashHugFace/engine-failure-prediction-model", filename="best_model.pkl")
model = joblib.load(path)
predictions = model.predict(X_new)
```