8 lines
400 B
Python
8 lines
400 B
Python
# Example usage
|
|
from scripts.run_assessment_prediction_trainer import CompanyModelPipeline
|
|
company_ids = ['company_123', 'company_456', 'company_789']
|
|
input_base_path = '/root/ds_erp_ai/data/raw/dummy_assessment_data.csv' # The base path where the raw data for each company is stored
|
|
|
|
pipeline = CompanyModelPipeline(company_ids=company_ids, input_base_path=input_base_path)
|
|
pipeline.run_pipeline()
|