SIENTIAPDE-1646
Update MLFlow reference artifact candidates to include 'test_data.csv' instead of 'train_data.csv'
This commit is contained in:
@@ -53,7 +53,7 @@ class MLFlow(SientiaMonitoring):
|
|||||||
|
|
||||||
_MAX_DEBUG_DATAFRAME_ROWS = 100
|
_MAX_DEBUG_DATAFRAME_ROWS = 100
|
||||||
_DEFAULT_MODEL_ALIAS = 'production'
|
_DEFAULT_MODEL_ALIAS = 'production'
|
||||||
_REFERENCE_ARTIFACT_CANDIDATES = ('retrain_input.csv', 'train_data.csv')
|
_REFERENCE_ARTIFACT_CANDIDATES = ('retrain_input.csv', 'test_data.csv')
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
|||||||
Reference in New Issue
Block a user