ML Pipeline
Machine learning for scientific research. Venv: source /Users/zhangmingda/clawd/.venv/bin/activate
Pipeline Overview
Data → Clean → Features → Split → Train → Evaluate → Interpret → Report
Model Selection Guide
| Task |
Data Size |
Interpretability Need |
Recommended |
| Classification (small) |
< 10K |
High |
Logistic Regression, Decision Tree |
| Classification (medium) |
10K-100K |
Medium |
Random Forest, XGBoost |
| Classification (large) |
> 100K |
Low OK |
Neural Network, XGBoost |
| Regression (linear) |
Any |
High |
Linear/Ridge/Lasso |
| Regression (nonlinear) |
Medium+ |
Medium |
Random Forest, Gradient Boosting |
| Clustering |
Any |
Medium |
K-Means, DBSCAN, Hierarchical |
| Dimensionality reduction |
Any |
Medium |
PCA, t-SNE, UMAP |
| Anomaly detection |
Any |
Medium |
Isolation Forest, LOF |
| Time series |
Any |
Varies |
ARIMA, Prophet, LSTM |
Standard Pipeline
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split, cross_val_score, GridSearchCV
from sklearn.preprocessing import StandardScaler, LabelEncoder
from sklearn.metrics import classification_report, confusion_matrix, roc_auc_score
from sklearn.ensemble import RandomForestClassifier, GradientBoostingClassifier
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import Pipeline
# 1. Preprocessing
X = df.drop('target', axis=1)
y = df['target']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42, stratify=y)
# 2. Pipeline with scaling
pipe = Pipeline([
('scaler', StandardScaler()),
('model', RandomForestClassifier(random_state=42))
])
# 3. Cross-validation
scores = cross_val_score(pipe, X_train, y_train, cv=5, scoring='roc_auc')
print(f"CV AUC: {scores.mean():.3f} ± {scores.std():.3f}")
# 4. Hyperparameter tuning
param_grid = {
'model__n_estimators': [100, 300, 500],
'model__max_depth': [5, 10, None],
'model__min_samples_leaf': [1, 5, 10]
}
grid = GridSearchCV(pipe, param_grid, cv=5, scoring='roc_auc', n_jobs=-1)
grid.fit(X_train, y_train)
# 5. Evaluation
y_pred = grid.predict(X_test)
print(classification_report(y_test, y_pred))
print(f"Test AUC: {roc_auc_score(y_test, grid.predict_proba(X_test)[:,1]):.3f}")
Feature Importance & Explainability
# Built-in importance (tree models)
importances = grid.best_estimator_.named_steps['model'].feature_importances_
feat_imp = pd.Series(importances, index=X.columns).sort_values(ascending=False)
# SHAP values (model-agnostic)
# pip install shap
import shap
explainer = shap.TreeExplainer(model)
shap_values = explainer.shap_values(X_test)
shap.summary_plot(shap_values, X_test)
Unsupervised Learning
from sklearn.cluster import KMeans, DBSCAN
from sklearn.decomposition import PCA
from sklearn.manifold import TSNE
# PCA
pca = PCA(n_components=0.95) # retain 95% variance
X_pca = pca.fit_transform(X_scaled)
print(f"Components: {pca.n_components_}, Explained variance: {pca.explained_variance_ratio_.cumsum()[-1]:.3f}")
# K-Means with elbow method
inertias = [KMeans(n_clusters=k, random_state=42).fit(X_scaled).inertia_ for k in range(2, 11)]
# t-SNE visualization
X_tsne = TSNE(n_components=2, random_state=42, perplexity=30).fit_transform(X_scaled)
Reporting ML Results in Papers
Always include:
- Dataset description (size, features, class balance)
- Preprocessing steps
- Model selection rationale
- Cross-validation strategy (k-fold, stratified, leave-one-out)
- Hyperparameter search space and method
- Multiple metrics (accuracy, precision, recall, F1, AUC)
- Comparison with baselines
- Feature importance / model interpretation
- Confidence intervals or statistical tests on performance
- Code/data availability statement
Tips
- Always use stratified splits for imbalanced data
- Report multiple metrics, not just accuracy
- Compare against simple baselines (majority class, mean prediction)
- Use nested CV for unbiased performance estimation
- Check for data leakage (especially with time series)
- Document random seeds for reproducibility
1---2name: ml-pipeline3description: Machine learning pipeline for scientific research including data preprocessing, feature engineering, model selection, training, evaluation, and interpretation. Covers supervised/unsupervised learning, deep learning, cross-validation, hyperparameter tuning, and model explainability. Use when user asks to build a predictive model, classify data, cluster samples, do feature selection, or apply ML to research data. Triggers on "machine learning", "classification", "clustering", "random forest", "neural network", "deep learning", "predict", "feature selection", "cross-validation", "train model".4---56# ML Pipeline78Machine learning for scientific research. Venv: `source /Users/zhangmingda/clawd/.venv/bin/activate`910## Pipeline Overview1112```13Data → Clean → Features → Split → Train → Evaluate → Interpret → Report14```1516## Model Selection Guide1718| Task | Data Size | Interpretability Need | Recommended |19|------|-----------|----------------------|-------------|20| Classification (small) | < 10K | High | Logistic Regression, Decision Tree |21| Classification (medium) | 10K-100K | Medium | Random Forest, XGBoost |22| Classification (large) | > 100K | Low OK | Neural Network, XGBoost |23| Regression (linear) | Any | High | Linear/Ridge/Lasso |24| Regression (nonlinear) | Medium+ | Medium | Random Forest, Gradient Boosting |25| Clustering | Any | Medium | K-Means, DBSCAN, Hierarchical |26| Dimensionality reduction | Any | Medium | PCA, t-SNE, UMAP |27| Anomaly detection | Any | Medium | Isolation Forest, LOF |28| Time series | Any | Varies | ARIMA, Prophet, LSTM |2930## Standard Pipeline3132```python33import numpy as np34import pandas as pd35from sklearn.model_selection import train_test_split, cross_val_score, GridSearchCV36from sklearn.preprocessing import StandardScaler, LabelEncoder37from sklearn.metrics import classification_report, confusion_matrix, roc_auc_score38from sklearn.ensemble import RandomForestClassifier, GradientBoostingClassifier39from sklearn.linear_model import LogisticRegression40from sklearn.pipeline import Pipeline4142# 1. Preprocessing43X = df.drop('target', axis=1)44y = df['target']45X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42, stratify=y)4647# 2. Pipeline with scaling48pipe = Pipeline([49 ('scaler', StandardScaler()),50 ('model', RandomForestClassifier(random_state=42))51])5253# 3. Cross-validation54scores = cross_val_score(pipe, X_train, y_train, cv=5, scoring='roc_auc')55print(f"CV AUC: {scores.mean():.3f} ± {scores.std():.3f}")5657# 4. Hyperparameter tuning58param_grid = {59 'model__n_estimators': [100, 300, 500],60 'model__max_depth': [5, 10, None],61 'model__min_samples_leaf': [1, 5, 10]62}63grid = GridSearchCV(pipe, param_grid, cv=5, scoring='roc_auc', n_jobs=-1)64grid.fit(X_train, y_train)6566# 5. Evaluation67y_pred = grid.predict(X_test)68print(classification_report(y_test, y_pred))69print(f"Test AUC: {roc_auc_score(y_test, grid.predict_proba(X_test)[:,1]):.3f}")70```7172## Feature Importance & Explainability7374```python75# Built-in importance (tree models)76importances = grid.best_estimator_.named_steps['model'].feature_importances_77feat_imp = pd.Series(importances, index=X.columns).sort_values(ascending=False)7879# SHAP values (model-agnostic)80# pip install shap81import shap82explainer = shap.TreeExplainer(model)83shap_values = explainer.shap_values(X_test)84shap.summary_plot(shap_values, X_test)85```8687## Unsupervised Learning8889```python90from sklearn.cluster import KMeans, DBSCAN91from sklearn.decomposition import PCA92from sklearn.manifold import TSNE9394# PCA95pca = PCA(n_components=0.95) # retain 95% variance96X_pca = pca.fit_transform(X_scaled)97print(f"Components: {pca.n_components_}, Explained variance: {pca.explained_variance_ratio_.cumsum()[-1]:.3f}")9899# K-Means with elbow method100inertias = [KMeans(n_clusters=k, random_state=42).fit(X_scaled).inertia_ for k in range(2, 11)]101102# t-SNE visualization103X_tsne = TSNE(n_components=2, random_state=42, perplexity=30).fit_transform(X_scaled)104```105106## Reporting ML Results in Papers107108Always include:1091. Dataset description (size, features, class balance)1102. Preprocessing steps1113. Model selection rationale1124. Cross-validation strategy (k-fold, stratified, leave-one-out)1135. Hyperparameter search space and method1146. Multiple metrics (accuracy, precision, recall, F1, AUC)1157. Comparison with baselines1168. Feature importance / model interpretation1179. Confidence intervals or statistical tests on performance11810. Code/data availability statement119120## Tips121- Always use stratified splits for imbalanced data122- Report multiple metrics, not just accuracy123- Compare against simple baselines (majority class, mean prediction)124- Use nested CV for unbiased performance estimation125- Check for data leakage (especially with time series)126- Document random seeds for reproducibility