refactor: update input dataset and training parameters

- Renamed columns in `input_dataset.csv` from `feature_a`, `feature_b`, and `target` to `Counter`, `Rollout`, and `Square` for better clarity.
- Updated the `run_training_test.py` script to reflect the new column names in the workflow input, ensuring consistency in data processing.
- Added `date_column` parameter to the workflow input for improved data handling.
This commit is contained in:
vitor-aignosi
2026-05-11 12:49:04 -03:00
parent 544c768ef4
commit bcd4fab259
2 changed files with 6 additions and 4 deletions

View File

@@ -87,6 +87,7 @@ request_data = {
'data_model_kwargs': {},
'model_kwargs': {},
'opt_params': {},
'date_column': 'timestamp',
'model_metadata': {'schemas': {'components': {'schemas': {}}}},
}
@@ -133,8 +134,8 @@ TH, TN, TQ = TEMPORAL_HOST, TEMPORAL_NAMESPACE, TRAIN_TASK_QUEUE
_workflow_input = {
'experiment_run_id': EXPERIMENT_RUN_ID,
'variable_columns': ['feature_a', 'feature_b'],
'target_variable': 'target',
'variable_columns': ['Counter', 'Rollout'],
'target_variable': 'Square',
'bucket_name': MINIO_BUCKET,
'file_name': OBJECT_NAME,
'line_separator': ',',
@@ -142,12 +143,13 @@ _workflow_input = {
'train_size': 80,
'shuffle': True,
'random_state': 42,
'model_name': 'test-runtime-linear-regression-model',
'model_name': 'test-runtime',
'model_type': 'linear_regression',
'model_id': EXPERIMENT_RUN_ID,
'data_model_kwargs': {},
'model_kwargs': {},
'opt_params': {},
'date_column': 'timestamp',
}
c = await client.Client.connect(