Decision Tree Classification of Transformer Condition from Dissolved Gas Analysis Results

Project info

• Field of studyElectrical Engineering and Energy
• LevelUndergraduate thesis (S1)
• Method typeClassification
• Estimated build time± 3 working days
• Toolsscikit-learn DecisionTreeClassifier, plot_tree, export_text
• PriceUS$16 - US$106

Research goal

Classify transformer condition using the features hydrogen gas and acetylene gas with Decision Tree and assess its accuracy.

Variables

Example result

• 0.991Accuracy
• 0.982Precision
• 1Recall
• 0.991F1-score
• 4Tree depth
• 10Leaves

The decision tree model classifies Transformer condition (low/high) from Hydrogen gas and Acetylene gas with test accuracy 0.991, F1 0.991 and AUC 0.988. Tree depth is 4 with 10 leaves.

Computed from illustrative synthetic data, not from a real study.

About Decision Tree

A decision tree recursively splits the data with if-then rules that are easy to read and visualise. It suits classification whose results must be explained as rules to non-technical readers.

Steps

1. 1Split into stratified training and test sets
2. 2Train DecisionTreeClassifier with a max_depth limit
3. 3Prune (ccp_alpha or min_samples_leaf) to reduce overfitting
4. 4Evaluate accuracy and F1 on the test set
5. 5Plot the tree and read its rules
6. 6Compute feature importance

Assumptions to check

• Deep trees overfit easily, so depth must be limited
• Results are sensitive to small changes in the data
• Classes should not be heavily imbalanced

Reading and reporting the results

Show the tree diagram and translate its main branches into meaningful rules, together with test accuracy. Mention that a single tree is less stable than ensembles.

Example Python code

A short starting script with this project's variable names. The delivered version is adapted to your data and fully commented.

import numpy as np
import matplotlib.pyplot as plt
from sklearn.tree import DecisionTreeClassifier, plot_tree, export_text
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, f1_score

rng = np.random.default_rng(21)
n = 300
hydrogen_gas = rng.normal(0, 1, n)
acetylene_gas = rng.normal(0, 1, n)
transformer_condition = ((hydrogen_gas > 0.2) & (acetylene_gas < 0.5)).astype(int)
flip = rng.random(n) < 0.05
transformer_condition[flip] = 1 - transformer_condition[flip]

X = np.column_stack([hydrogen_gas, acetylene_gas])
X_tr, X_te, y_tr, y_te = train_test_split(X, transformer_condition, test_size=0.3, stratify=transformer_condition, random_state=21)
tree = DecisionTreeClassifier(max_depth=3, min_samples_leaf=5, random_state=21)
tree.fit(X_tr, y_tr)
pred = tree.predict(X_te)
print("Accuracy:", round(accuracy_score(y_te, pred), 3))
print("F1:", round(f1_score(y_te, pred), 3))
print("Importance:", tree.feature_importances_.round(3))
print(export_text(tree, feature_names=["x1", "x2"]))

plt.figure(figsize=(9, 5))
plot_tree(tree, feature_names=["x1", "x2"], class_names=["0", "1"], filled=True)
plt.show()

Packages and prices

Choose how much help you need. Each higher package includes everything in the lower ones. Prices exclude 11% VAT.

• Payment: 50% down payment before work starts, balance after the deliverables are handed over
• Tax: Prices exclude 11% VAT
• Data and academic integrity: Results are computed from your own research data; you write and stand behind your manuscript, we help with the analysis and your understanding of it
• Revisions: Up to 2 revisions are included in the price (Package 3: 3 rounds)

Frequently asked questions