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.
- autoforge_engine-0.1.0/LICENSE +0 -0
- autoforge_engine-0.1.0/PKG-INFO +105 -0
- autoforge_engine-0.1.0/README.md +78 -0
- autoforge_engine-0.1.0/autoforge_engine.egg-info/PKG-INFO +105 -0
- autoforge_engine-0.1.0/autoforge_engine.egg-info/SOURCES.txt +72 -0
- autoforge_engine-0.1.0/autoforge_engine.egg-info/dependency_links.txt +1 -0
- autoforge_engine-0.1.0/autoforge_engine.egg-info/entry_points.txt +2 -0
- autoforge_engine-0.1.0/autoforge_engine.egg-info/requires.txt +20 -0
- autoforge_engine-0.1.0/autoforge_engine.egg-info/top_level.txt +1 -0
- autoforge_engine-0.1.0/modelforge/artifact_manager.py +485 -0
- autoforge_engine-0.1.0/modelforge/automl.py +1472 -0
- autoforge_engine-0.1.0/modelforge/cli.py +1258 -0
- autoforge_engine-0.1.0/modelforge/column_intelligence.py +404 -0
- autoforge_engine-0.1.0/modelforge/config.py +580 -0
- autoforge_engine-0.1.0/modelforge/cross_validation.py +749 -0
- autoforge_engine-0.1.0/modelforge/data_audit.py +392 -0
- autoforge_engine-0.1.0/modelforge/data_loader.py +76 -0
- autoforge_engine-0.1.0/modelforge/evaluation.py +397 -0
- autoforge_engine-0.1.0/modelforge/experiment_tracker.py +490 -0
- autoforge_engine-0.1.0/modelforge/explainability.py +346 -0
- autoforge_engine-0.1.0/modelforge/feature_engineering.py +393 -0
- autoforge_engine-0.1.0/modelforge/feature_selection.py +528 -0
- autoforge_engine-0.1.0/modelforge/hyperparameter_optimization.py +593 -0
- autoforge_engine-0.1.0/modelforge/model_registry.py +684 -0
- autoforge_engine-0.1.0/modelforge/model_screening.py +531 -0
- autoforge_engine-0.1.0/modelforge/persistence.py +456 -0
- autoforge_engine-0.1.0/modelforge/pipeline_generator.py +278 -0
- autoforge_engine-0.1.0/modelforge/prediction_validator.py +316 -0
- autoforge_engine-0.1.0/modelforge/preprocessing.py +179 -0
- autoforge_engine-0.1.0/modelforge/profiler.py +85 -0
- autoforge_engine-0.1.0/modelforge/ranking.py +351 -0
- autoforge_engine-0.1.0/modelforge/reproducibility.py +295 -0
- autoforge_engine-0.1.0/modelforge/reproducibility_integration.py +192 -0
- autoforge_engine-0.1.0/modelforge/run_manager.py +200 -0
- autoforge_engine-0.1.0/modelforge/target_selector.py +108 -0
- autoforge_engine-0.1.0/pyproject.toml +51 -0
- autoforge_engine-0.1.0/setup.cfg +4 -0
- autoforge_engine-0.1.0/tests/test_artifact_manager.py +413 -0
- autoforge_engine-0.1.0/tests/test_automl.py +852 -0
- autoforge_engine-0.1.0/tests/test_automl_reproducibility.py +495 -0
- autoforge_engine-0.1.0/tests/test_automl_robustness.py +179 -0
- autoforge_engine-0.1.0/tests/test_cli.py +593 -0
- autoforge_engine-0.1.0/tests/test_cli_workflow.py +355 -0
- autoforge_engine-0.1.0/tests/test_column_intelligence.py +135 -0
- autoforge_engine-0.1.0/tests/test_config.py +386 -0
- autoforge_engine-0.1.0/tests/test_config_automl_integration.py +278 -0
- autoforge_engine-0.1.0/tests/test_cross_validation.py +431 -0
- autoforge_engine-0.1.0/tests/test_data_audit.py +153 -0
- autoforge_engine-0.1.0/tests/test_data_loader.py +46 -0
- autoforge_engine-0.1.0/tests/test_end_to_end.py +404 -0
- autoforge_engine-0.1.0/tests/test_evaluation.py +339 -0
- autoforge_engine-0.1.0/tests/test_experiment_artifacts.py +579 -0
- autoforge_engine-0.1.0/tests/test_experiment_tracker.py +311 -0
- autoforge_engine-0.1.0/tests/test_experiment_tracker_reproducibility.py +250 -0
- autoforge_engine-0.1.0/tests/test_explainability.py +424 -0
- autoforge_engine-0.1.0/tests/test_failure_isolation.py +252 -0
- autoforge_engine-0.1.0/tests/test_feature_engineering.py +217 -0
- autoforge_engine-0.1.0/tests/test_feature_selection.py +219 -0
- autoforge_engine-0.1.0/tests/test_full_system.py +757 -0
- autoforge_engine-0.1.0/tests/test_hyperparameter_optimization.py +360 -0
- autoforge_engine-0.1.0/tests/test_model_registry.py +232 -0
- autoforge_engine-0.1.0/tests/test_model_screening.py +617 -0
- autoforge_engine-0.1.0/tests/test_persistence.py +777 -0
- autoforge_engine-0.1.0/tests/test_pipeline_generator.py +386 -0
- autoforge_engine-0.1.0/tests/test_prediction_robustness.py +274 -0
- autoforge_engine-0.1.0/tests/test_prediction_validator.py +340 -0
- autoforge_engine-0.1.0/tests/test_preprocessing.py +193 -0
- autoforge_engine-0.1.0/tests/test_profiler.py +75 -0
- autoforge_engine-0.1.0/tests/test_ranking.py +446 -0
- autoforge_engine-0.1.0/tests/test_reproducibility.py +302 -0
- autoforge_engine-0.1.0/tests/test_reproducibility_integration.py +303 -0
- autoforge_engine-0.1.0/tests/test_robustness.py +132 -0
- autoforge_engine-0.1.0/tests/test_run_manager.py +227 -0
- 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 @@
|
|
|
1
|
+
|
|
@@ -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
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
modelforge
|