Skip to content

Commit 403f318

Browse files
author
Stephen Hoover
authored
ENH Add CivisML v2.2 template IDs (#247)
The ``ModelPipeline`` will now create v2.2 models in CivisML by default.
1 parent d3170cb commit 403f318

File tree

3 files changed

+8
-4
lines changed

3 files changed

+8
-4
lines changed

CHANGELOG.md

+3
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ This project adheres to [Semantic Versioning](http://semver.org/).
2121
- Executors in ``futures`` (and the joblib backend, which uses them) will now
2222
add "CIVIS_PARENT_JOB_ID" and "CIVIS_PARENT_RUN_ID" environment variables
2323
to the child jobs they create (#236)
24+
- Update default CivisML version to v2.2. This includes a new function
25+
``ModelPipeline.register_pretrained_model`` which allows users to train
26+
a model outside of Civis Platform and use CivisML to score it at scale (#242, #247).
2427
- Added a new parameter ``dvs_to_predict`` to ``civis.ml.ModelPipeline.predict``.
2528
This allows users to select a subset of a model's outputs for scoring (#241).
2629
- Added `civis.io.export_to_civis_file` to store results of a SQL query

civis/ml/_model.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,11 @@
4242
9112: 9113, # v1.1
4343
8387: 9113, # v1.0
4444
7020: 7021, # v0.5
45-
11028: 10616, # v2.2 registration CHANGE ME
45+
11219: 11220, # v2.2
46+
11221: 11220, # v2.2 registration
4647
}
4748
_CIVISML_TEMPLATE = None # CivisML training template to use
48-
REGISTRATION_TEMPLATES = [11028, # v2.2 CHANGE ME
49+
REGISTRATION_TEMPLATES = [11221, # v2.2
4950
]
5051

5152

civis/ml/tests/test_model.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@
3939
from civis.ml import _model
4040

4141

42-
LATEST_TRAIN_TEMPLATE = 10582
43-
LATEST_PRED_TEMPLATE = 10583
42+
LATEST_TRAIN_TEMPLATE = 11219
43+
LATEST_PRED_TEMPLATE = 11220
4444

4545

4646
def setup_client_mock(script_id=-10, run_id=100, state='succeeded',

0 commit comments

Comments
 (0)