autoforge-engine 0.1.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (74) hide show
  1. autoforge_engine-0.1.0/LICENSE +0 -0
  2. autoforge_engine-0.1.0/PKG-INFO +105 -0
  3. autoforge_engine-0.1.0/README.md +78 -0
  4. autoforge_engine-0.1.0/autoforge_engine.egg-info/PKG-INFO +105 -0
  5. autoforge_engine-0.1.0/autoforge_engine.egg-info/SOURCES.txt +72 -0
  6. autoforge_engine-0.1.0/autoforge_engine.egg-info/dependency_links.txt +1 -0
  7. autoforge_engine-0.1.0/autoforge_engine.egg-info/entry_points.txt +2 -0
  8. autoforge_engine-0.1.0/autoforge_engine.egg-info/requires.txt +20 -0
  9. autoforge_engine-0.1.0/autoforge_engine.egg-info/top_level.txt +1 -0
  10. autoforge_engine-0.1.0/modelforge/artifact_manager.py +485 -0
  11. autoforge_engine-0.1.0/modelforge/automl.py +1472 -0
  12. autoforge_engine-0.1.0/modelforge/cli.py +1258 -0
  13. autoforge_engine-0.1.0/modelforge/column_intelligence.py +404 -0
  14. autoforge_engine-0.1.0/modelforge/config.py +580 -0
  15. autoforge_engine-0.1.0/modelforge/cross_validation.py +749 -0
  16. autoforge_engine-0.1.0/modelforge/data_audit.py +392 -0
  17. autoforge_engine-0.1.0/modelforge/data_loader.py +76 -0
  18. autoforge_engine-0.1.0/modelforge/evaluation.py +397 -0
  19. autoforge_engine-0.1.0/modelforge/experiment_tracker.py +490 -0
  20. autoforge_engine-0.1.0/modelforge/explainability.py +346 -0
  21. autoforge_engine-0.1.0/modelforge/feature_engineering.py +393 -0
  22. autoforge_engine-0.1.0/modelforge/feature_selection.py +528 -0
  23. autoforge_engine-0.1.0/modelforge/hyperparameter_optimization.py +593 -0
  24. autoforge_engine-0.1.0/modelforge/model_registry.py +684 -0
  25. autoforge_engine-0.1.0/modelforge/model_screening.py +531 -0
  26. autoforge_engine-0.1.0/modelforge/persistence.py +456 -0
  27. autoforge_engine-0.1.0/modelforge/pipeline_generator.py +278 -0
  28. autoforge_engine-0.1.0/modelforge/prediction_validator.py +316 -0
  29. autoforge_engine-0.1.0/modelforge/preprocessing.py +179 -0
  30. autoforge_engine-0.1.0/modelforge/profiler.py +85 -0
  31. autoforge_engine-0.1.0/modelforge/ranking.py +351 -0
  32. autoforge_engine-0.1.0/modelforge/reproducibility.py +295 -0
  33. autoforge_engine-0.1.0/modelforge/reproducibility_integration.py +192 -0
  34. autoforge_engine-0.1.0/modelforge/run_manager.py +200 -0
  35. autoforge_engine-0.1.0/modelforge/target_selector.py +108 -0
  36. autoforge_engine-0.1.0/pyproject.toml +51 -0
  37. autoforge_engine-0.1.0/setup.cfg +4 -0
  38. autoforge_engine-0.1.0/tests/test_artifact_manager.py +413 -0
  39. autoforge_engine-0.1.0/tests/test_automl.py +852 -0
  40. autoforge_engine-0.1.0/tests/test_automl_reproducibility.py +495 -0
  41. autoforge_engine-0.1.0/tests/test_automl_robustness.py +179 -0
  42. autoforge_engine-0.1.0/tests/test_cli.py +593 -0
  43. autoforge_engine-0.1.0/tests/test_cli_workflow.py +355 -0
  44. autoforge_engine-0.1.0/tests/test_column_intelligence.py +135 -0
  45. autoforge_engine-0.1.0/tests/test_config.py +386 -0
  46. autoforge_engine-0.1.0/tests/test_config_automl_integration.py +278 -0
  47. autoforge_engine-0.1.0/tests/test_cross_validation.py +431 -0
  48. autoforge_engine-0.1.0/tests/test_data_audit.py +153 -0
  49. autoforge_engine-0.1.0/tests/test_data_loader.py +46 -0
  50. autoforge_engine-0.1.0/tests/test_end_to_end.py +404 -0
  51. autoforge_engine-0.1.0/tests/test_evaluation.py +339 -0
  52. autoforge_engine-0.1.0/tests/test_experiment_artifacts.py +579 -0
  53. autoforge_engine-0.1.0/tests/test_experiment_tracker.py +311 -0
  54. autoforge_engine-0.1.0/tests/test_experiment_tracker_reproducibility.py +250 -0
  55. autoforge_engine-0.1.0/tests/test_explainability.py +424 -0
  56. autoforge_engine-0.1.0/tests/test_failure_isolation.py +252 -0
  57. autoforge_engine-0.1.0/tests/test_feature_engineering.py +217 -0
  58. autoforge_engine-0.1.0/tests/test_feature_selection.py +219 -0
  59. autoforge_engine-0.1.0/tests/test_full_system.py +757 -0
  60. autoforge_engine-0.1.0/tests/test_hyperparameter_optimization.py +360 -0
  61. autoforge_engine-0.1.0/tests/test_model_registry.py +232 -0
  62. autoforge_engine-0.1.0/tests/test_model_screening.py +617 -0
  63. autoforge_engine-0.1.0/tests/test_persistence.py +777 -0
  64. autoforge_engine-0.1.0/tests/test_pipeline_generator.py +386 -0
  65. autoforge_engine-0.1.0/tests/test_prediction_robustness.py +274 -0
  66. autoforge_engine-0.1.0/tests/test_prediction_validator.py +340 -0
  67. autoforge_engine-0.1.0/tests/test_preprocessing.py +193 -0
  68. autoforge_engine-0.1.0/tests/test_profiler.py +75 -0
  69. autoforge_engine-0.1.0/tests/test_ranking.py +446 -0
  70. autoforge_engine-0.1.0/tests/test_reproducibility.py +302 -0
  71. autoforge_engine-0.1.0/tests/test_reproducibility_integration.py +303 -0
  72. autoforge_engine-0.1.0/tests/test_robustness.py +132 -0
  73. autoforge_engine-0.1.0/tests/test_run_manager.py +227 -0
  74. autoforge_engine-0.1.0/tests/test_target_selector.py +87 -0
File without changes
@@ -0,0 +1,105 @@
1
+ Metadata-Version: 2.4
2
+ Name: autoforge-engine
3
+ Version: 0.1.0
4
+ Summary: A transparent, local-first AutoML framework for automated ML pipeline discovery.
5
+ Author: Aditya Kumar Singh
6
+ Requires-Python: >=3.11
7
+ Description-Content-Type: text/markdown
8
+ License-File: LICENSE
9
+ Requires-Dist: numpy>=1.26
10
+ Requires-Dist: pandas>=2.1
11
+ Requires-Dist: scikit-learn>=1.4
12
+ Requires-Dist: rich>=13.7
13
+ Requires-Dist: typer>=0.12
14
+ Requires-Dist: pyyaml>=6.0
15
+ Provides-Extra: boosting
16
+ Requires-Dist: xgboost>=2.0; extra == "boosting"
17
+ Requires-Dist: lightgbm>=4.0; extra == "boosting"
18
+ Requires-Dist: catboost>=1.2; extra == "boosting"
19
+ Provides-Extra: optimization
20
+ Requires-Dist: optuna>=3.6; extra == "optimization"
21
+ Provides-Extra: dev
22
+ Requires-Dist: pytest>=8.0; extra == "dev"
23
+ Requires-Dist: ruff>=0.5; extra == "dev"
24
+ Requires-Dist: black>=24.0; extra == "dev"
25
+ Requires-Dist: mypy>=1.10; extra == "dev"
26
+ Dynamic: license-file
27
+
28
+ # ModelForge
29
+
30
+ ModelForge is a transparent, local-first AutoML framework for discovering,
31
+ evaluating, explaining, and persisting scikit-learn pipelines. It keeps the
32
+ workflow inspectable: data audits, preprocessing, model screening,
33
+ cross-validation, ranking, reproducibility metadata, and artifacts are all
34
+ available as Python objects and local files.
35
+
36
+ ## Quick start
37
+
38
+ ```bash
39
+ python -m pip install -e ".[dev]"
40
+ modelforge train --data examples/house_prices.csv --target price \
41
+ --task-type regression --output model.joblib
42
+ modelforge predict --model model.joblib --data examples/house_prices_new.csv
43
+ ```
44
+
45
+ The same workflow is available from Python:
46
+
47
+ ```python
48
+ from modelforge import AutoML
49
+
50
+ automl = AutoML(cv=5, random_state=42)
51
+ result = automl.fit(data, target="price", task_type="regression")
52
+ automl.save("artifacts/house-prices.joblib", overwrite=True)
53
+ print(result["best_model"])
54
+ ```
55
+
56
+ ## What is included
57
+
58
+ ModelForge currently provides ingestion, target detection, profiling, column
59
+ intelligence, data-quality and leakage auditing, preprocessing, feature
60
+ engineering and selection, a model registry, screening, cross-validation,
61
+ ranking, explainability, persistence, prediction validation, optimization,
62
+ experiment tracking, run management, configuration, CLI workflows, failure
63
+ isolation, robustness checks, and reproducibility snapshots.
64
+
65
+ Optional model and optimization integrations are declared in `pyproject.toml`:
66
+
67
+ ```bash
68
+ python -m pip install -e ".[boosting,optimization]"
69
+ ```
70
+
71
+ ## CLI
72
+
73
+ ```bash
74
+ modelforge --help
75
+ modelforge models --task-type classification
76
+ modelforge experiments list --directory .modelforge/experiments
77
+ modelforge experiments get EXPERIMENT_ID
78
+ modelforge experiments compare ID_ONE ID_TWO
79
+ ```
80
+
81
+ The CLI uses Rich tables and keeps generated models, experiment JSON, and
82
+ metadata local by default.
83
+
84
+ ## Examples and benchmarks
85
+
86
+ - `examples/end_to_end_regression.py` demonstrates a complete Python workflow.
87
+ - `examples/end_to_end_classification.py` demonstrates classification and
88
+ probability prediction.
89
+ - `benchmarks/benchmark_baselines.py` compares ModelForge with simple baseline
90
+ estimators and records runtime and quality metrics.
91
+
92
+ Run an example with `python examples/end_to_end_regression.py` or run the
93
+ benchmark with `python benchmarks/benchmark_baselines.py`.
94
+
95
+ ## Development
96
+
97
+ ```bash
98
+ .venv\\Scripts\\python.exe -m pytest -q
99
+ ruff check .
100
+ ```
101
+
102
+ GitHub Actions runs the test suite on supported Python versions. The roadmap
103
+ for the remaining product work lives in `docs/ROADMAP.md`; it separates
104
+ implemented foundations from integrations that require optional dependencies
105
+ or additional design work.
@@ -0,0 +1,78 @@
1
+ # ModelForge
2
+
3
+ ModelForge is a transparent, local-first AutoML framework for discovering,
4
+ evaluating, explaining, and persisting scikit-learn pipelines. It keeps the
5
+ workflow inspectable: data audits, preprocessing, model screening,
6
+ cross-validation, ranking, reproducibility metadata, and artifacts are all
7
+ available as Python objects and local files.
8
+
9
+ ## Quick start
10
+
11
+ ```bash
12
+ python -m pip install -e ".[dev]"
13
+ modelforge train --data examples/house_prices.csv --target price \
14
+ --task-type regression --output model.joblib
15
+ modelforge predict --model model.joblib --data examples/house_prices_new.csv
16
+ ```
17
+
18
+ The same workflow is available from Python:
19
+
20
+ ```python
21
+ from modelforge import AutoML
22
+
23
+ automl = AutoML(cv=5, random_state=42)
24
+ result = automl.fit(data, target="price", task_type="regression")
25
+ automl.save("artifacts/house-prices.joblib", overwrite=True)
26
+ print(result["best_model"])
27
+ ```
28
+
29
+ ## What is included
30
+
31
+ ModelForge currently provides ingestion, target detection, profiling, column
32
+ intelligence, data-quality and leakage auditing, preprocessing, feature
33
+ engineering and selection, a model registry, screening, cross-validation,
34
+ ranking, explainability, persistence, prediction validation, optimization,
35
+ experiment tracking, run management, configuration, CLI workflows, failure
36
+ isolation, robustness checks, and reproducibility snapshots.
37
+
38
+ Optional model and optimization integrations are declared in `pyproject.toml`:
39
+
40
+ ```bash
41
+ python -m pip install -e ".[boosting,optimization]"
42
+ ```
43
+
44
+ ## CLI
45
+
46
+ ```bash
47
+ modelforge --help
48
+ modelforge models --task-type classification
49
+ modelforge experiments list --directory .modelforge/experiments
50
+ modelforge experiments get EXPERIMENT_ID
51
+ modelforge experiments compare ID_ONE ID_TWO
52
+ ```
53
+
54
+ The CLI uses Rich tables and keeps generated models, experiment JSON, and
55
+ metadata local by default.
56
+
57
+ ## Examples and benchmarks
58
+
59
+ - `examples/end_to_end_regression.py` demonstrates a complete Python workflow.
60
+ - `examples/end_to_end_classification.py` demonstrates classification and
61
+ probability prediction.
62
+ - `benchmarks/benchmark_baselines.py` compares ModelForge with simple baseline
63
+ estimators and records runtime and quality metrics.
64
+
65
+ Run an example with `python examples/end_to_end_regression.py` or run the
66
+ benchmark with `python benchmarks/benchmark_baselines.py`.
67
+
68
+ ## Development
69
+
70
+ ```bash
71
+ .venv\\Scripts\\python.exe -m pytest -q
72
+ ruff check .
73
+ ```
74
+
75
+ GitHub Actions runs the test suite on supported Python versions. The roadmap
76
+ for the remaining product work lives in `docs/ROADMAP.md`; it separates
77
+ implemented foundations from integrations that require optional dependencies
78
+ or additional design work.
@@ -0,0 +1,105 @@
1
+ Metadata-Version: 2.4
2
+ Name: autoforge-engine
3
+ Version: 0.1.0
4
+ Summary: A transparent, local-first AutoML framework for automated ML pipeline discovery.
5
+ Author: Aditya Kumar Singh
6
+ Requires-Python: >=3.11
7
+ Description-Content-Type: text/markdown
8
+ License-File: LICENSE
9
+ Requires-Dist: numpy>=1.26
10
+ Requires-Dist: pandas>=2.1
11
+ Requires-Dist: scikit-learn>=1.4
12
+ Requires-Dist: rich>=13.7
13
+ Requires-Dist: typer>=0.12
14
+ Requires-Dist: pyyaml>=6.0
15
+ Provides-Extra: boosting
16
+ Requires-Dist: xgboost>=2.0; extra == "boosting"
17
+ Requires-Dist: lightgbm>=4.0; extra == "boosting"
18
+ Requires-Dist: catboost>=1.2; extra == "boosting"
19
+ Provides-Extra: optimization
20
+ Requires-Dist: optuna>=3.6; extra == "optimization"
21
+ Provides-Extra: dev
22
+ Requires-Dist: pytest>=8.0; extra == "dev"
23
+ Requires-Dist: ruff>=0.5; extra == "dev"
24
+ Requires-Dist: black>=24.0; extra == "dev"
25
+ Requires-Dist: mypy>=1.10; extra == "dev"
26
+ Dynamic: license-file
27
+
28
+ # ModelForge
29
+
30
+ ModelForge is a transparent, local-first AutoML framework for discovering,
31
+ evaluating, explaining, and persisting scikit-learn pipelines. It keeps the
32
+ workflow inspectable: data audits, preprocessing, model screening,
33
+ cross-validation, ranking, reproducibility metadata, and artifacts are all
34
+ available as Python objects and local files.
35
+
36
+ ## Quick start
37
+
38
+ ```bash
39
+ python -m pip install -e ".[dev]"
40
+ modelforge train --data examples/house_prices.csv --target price \
41
+ --task-type regression --output model.joblib
42
+ modelforge predict --model model.joblib --data examples/house_prices_new.csv
43
+ ```
44
+
45
+ The same workflow is available from Python:
46
+
47
+ ```python
48
+ from modelforge import AutoML
49
+
50
+ automl = AutoML(cv=5, random_state=42)
51
+ result = automl.fit(data, target="price", task_type="regression")
52
+ automl.save("artifacts/house-prices.joblib", overwrite=True)
53
+ print(result["best_model"])
54
+ ```
55
+
56
+ ## What is included
57
+
58
+ ModelForge currently provides ingestion, target detection, profiling, column
59
+ intelligence, data-quality and leakage auditing, preprocessing, feature
60
+ engineering and selection, a model registry, screening, cross-validation,
61
+ ranking, explainability, persistence, prediction validation, optimization,
62
+ experiment tracking, run management, configuration, CLI workflows, failure
63
+ isolation, robustness checks, and reproducibility snapshots.
64
+
65
+ Optional model and optimization integrations are declared in `pyproject.toml`:
66
+
67
+ ```bash
68
+ python -m pip install -e ".[boosting,optimization]"
69
+ ```
70
+
71
+ ## CLI
72
+
73
+ ```bash
74
+ modelforge --help
75
+ modelforge models --task-type classification
76
+ modelforge experiments list --directory .modelforge/experiments
77
+ modelforge experiments get EXPERIMENT_ID
78
+ modelforge experiments compare ID_ONE ID_TWO
79
+ ```
80
+
81
+ The CLI uses Rich tables and keeps generated models, experiment JSON, and
82
+ metadata local by default.
83
+
84
+ ## Examples and benchmarks
85
+
86
+ - `examples/end_to_end_regression.py` demonstrates a complete Python workflow.
87
+ - `examples/end_to_end_classification.py` demonstrates classification and
88
+ probability prediction.
89
+ - `benchmarks/benchmark_baselines.py` compares ModelForge with simple baseline
90
+ estimators and records runtime and quality metrics.
91
+
92
+ Run an example with `python examples/end_to_end_regression.py` or run the
93
+ benchmark with `python benchmarks/benchmark_baselines.py`.
94
+
95
+ ## Development
96
+
97
+ ```bash
98
+ .venv\\Scripts\\python.exe -m pytest -q
99
+ ruff check .
100
+ ```
101
+
102
+ GitHub Actions runs the test suite on supported Python versions. The roadmap
103
+ for the remaining product work lives in `docs/ROADMAP.md`; it separates
104
+ implemented foundations from integrations that require optional dependencies
105
+ or additional design work.
@@ -0,0 +1,72 @@
1
+ LICENSE
2
+ README.md
3
+ pyproject.toml
4
+ autoforge_engine.egg-info/PKG-INFO
5
+ autoforge_engine.egg-info/SOURCES.txt
6
+ autoforge_engine.egg-info/dependency_links.txt
7
+ autoforge_engine.egg-info/entry_points.txt
8
+ autoforge_engine.egg-info/requires.txt
9
+ autoforge_engine.egg-info/top_level.txt
10
+ modelforge/artifact_manager.py
11
+ modelforge/automl.py
12
+ modelforge/cli.py
13
+ modelforge/column_intelligence.py
14
+ modelforge/config.py
15
+ modelforge/cross_validation.py
16
+ modelforge/data_audit.py
17
+ modelforge/data_loader.py
18
+ modelforge/evaluation.py
19
+ modelforge/experiment_tracker.py
20
+ modelforge/explainability.py
21
+ modelforge/feature_engineering.py
22
+ modelforge/feature_selection.py
23
+ modelforge/hyperparameter_optimization.py
24
+ modelforge/model_registry.py
25
+ modelforge/model_screening.py
26
+ modelforge/persistence.py
27
+ modelforge/pipeline_generator.py
28
+ modelforge/prediction_validator.py
29
+ modelforge/preprocessing.py
30
+ modelforge/profiler.py
31
+ modelforge/ranking.py
32
+ modelforge/reproducibility.py
33
+ modelforge/reproducibility_integration.py
34
+ modelforge/run_manager.py
35
+ modelforge/target_selector.py
36
+ tests/test_artifact_manager.py
37
+ tests/test_automl.py
38
+ tests/test_automl_reproducibility.py
39
+ tests/test_automl_robustness.py
40
+ tests/test_cli.py
41
+ tests/test_cli_workflow.py
42
+ tests/test_column_intelligence.py
43
+ tests/test_config.py
44
+ tests/test_config_automl_integration.py
45
+ tests/test_cross_validation.py
46
+ tests/test_data_audit.py
47
+ tests/test_data_loader.py
48
+ tests/test_end_to_end.py
49
+ tests/test_evaluation.py
50
+ tests/test_experiment_artifacts.py
51
+ tests/test_experiment_tracker.py
52
+ tests/test_experiment_tracker_reproducibility.py
53
+ tests/test_explainability.py
54
+ tests/test_failure_isolation.py
55
+ tests/test_feature_engineering.py
56
+ tests/test_feature_selection.py
57
+ tests/test_full_system.py
58
+ tests/test_hyperparameter_optimization.py
59
+ tests/test_model_registry.py
60
+ tests/test_model_screening.py
61
+ tests/test_persistence.py
62
+ tests/test_pipeline_generator.py
63
+ tests/test_prediction_robustness.py
64
+ tests/test_prediction_validator.py
65
+ tests/test_preprocessing.py
66
+ tests/test_profiler.py
67
+ tests/test_ranking.py
68
+ tests/test_reproducibility.py
69
+ tests/test_reproducibility_integration.py
70
+ tests/test_robustness.py
71
+ tests/test_run_manager.py
72
+ tests/test_target_selector.py
@@ -0,0 +1,2 @@
1
+ [console_scripts]
2
+ modelforge = modelforge.cli:app
@@ -0,0 +1,20 @@
1
+ numpy>=1.26
2
+ pandas>=2.1
3
+ scikit-learn>=1.4
4
+ rich>=13.7
5
+ typer>=0.12
6
+ pyyaml>=6.0
7
+
8
+ [boosting]
9
+ xgboost>=2.0
10
+ lightgbm>=4.0
11
+ catboost>=1.2
12
+
13
+ [dev]
14
+ pytest>=8.0
15
+ ruff>=0.5
16
+ black>=24.0
17
+ mypy>=1.10
18
+
19
+ [optimization]
20
+ optuna>=3.6