You can deploy this model and start earning money today!

vishal017/
steel-fault-classification-rf

Name: steel-fault-classification-rf
Availability: InStock

Views

No views yet

Steel Surface Fault Classification (Random Forest)

This repository contains a trained Random Forest model and its preprocessing pipeline for steel surface fault classification.

📦 Files

rf_fault_model.pkl – Trained Random Forest classifier
preprocessor.pkl – ColumnTransformer with RobustScaler

🧠 Model Overview

Algorithm: Random Forest Classifier
Target: Multiclass steel surface fault type
Preprocessing:
- Robust scaling for numerical features
- Binary features passed through unchanged
Hyperparameter tuning: GridSearchCV (5-fold, f1_macro)

🚀 How to Load and Use

python

import joblib
import pandas as pd

# Load files
model = joblib.load("rf_fault_model.pkl")
preprocessor = joblib.load("preprocessor.pkl")

# Example input (raw features)
sample = pd.DataFrame([{
    "X_Minimum": 42,
    "X_Maximum": 184,
    "Y_Minimum": 23,
    "Y_Maximum": 256,
    "TypeOfSteel_A300": 1,
    "TypeOfSteel_A400": 0
}])

# Preprocess + predict
sample_processed = preprocessor.transform(sample)
prediction = model.predict(sample_processed)

print("Predicted fault class:", prediction)

vishal017/steel-fault-classification-rf

Steel Surface Fault Classification (Random Forest)

📦 Files

🧠 Model Overview

🚀 How to Load and Use

vishal017/
steel-fault-classification-rf