chore: update .gitignore and requirements for development
- Added new entries to .gitignore to exclude temporary files and training input datasets, ensuring a cleaner repository. - Included type stubs for psycopg2 in requirements-dev.txt to enhance type checking support for database interactions. - Refactored the run_training_test.py script to implement a structured approach for loading and validating training input JSON files, improving the robustness of the training workflow.
This commit is contained in:
48
scripts/inputs/linear_regression.json
Normal file
48
scripts/inputs/linear_regression.json
Normal file
@@ -0,0 +1,48 @@
|
||||
{
|
||||
"experiment": {
|
||||
"experiment_run_id": 1001,
|
||||
"experiment_name": "test-experiment-name",
|
||||
"run_name": "test-run-name",
|
||||
"username": "vitor.santos@aignosi.com.br",
|
||||
"status": "ORCHESTRATOR_WAITING_PROC"
|
||||
},
|
||||
"minio": {
|
||||
"mc_alias": "suse",
|
||||
"bucket_name": "model-training",
|
||||
"file_name": "training-sample-dataset-1001.csv",
|
||||
"local_csv": "input_dataset.csv"
|
||||
},
|
||||
"temporal": {
|
||||
"task_queue": "train_model-basic-queue",
|
||||
"workflow_name": "train_model",
|
||||
"execution_timeout_minutes": 5,
|
||||
"run_timeout_minutes": 5,
|
||||
"task_timeout_minutes": 5
|
||||
},
|
||||
"payload": {
|
||||
"experiment_run_id": 1001,
|
||||
"variable_columns": ["Counter", "Rollout"],
|
||||
"target_variable": "Square",
|
||||
"line_separator": ",",
|
||||
"decimal_separator": ".",
|
||||
"train_size": 80,
|
||||
"shuffle": true,
|
||||
"random_state": 42,
|
||||
"model_name": "smoke-linreg",
|
||||
"model_type": "linear_regression",
|
||||
"model_id": 1001,
|
||||
"data_model_kwargs": {},
|
||||
"model_kwargs": {},
|
||||
"opt_params": {},
|
||||
"date_column": "timestamp"
|
||||
},
|
||||
"db_only": {
|
||||
"model_metadata": {
|
||||
"schemas": {
|
||||
"components": {
|
||||
"schemas": {}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user