2024-09-12 00:01:03 +00:00
|
|
|
# Example usage
|
2024-09-12 21:36:02 +00:00
|
|
|
'''from scripts.run_assessment_prediction_trainer import CompanyModelPipeline
|
|
|
|
|
company_ids = ['testid']
|
|
|
|
|
input_base_path = '/root/ds_erp_ai/data/raw/erp_assessment_prediction' # The base path where the raw data for each company is stored
|
2024-09-05 02:59:01 +00:00
|
|
|
|
2024-09-12 00:01:03 +00:00
|
|
|
pipeline = CompanyModelPipeline(company_ids=company_ids, input_base_path=input_base_path)
|
2024-09-12 21:36:02 +00:00
|
|
|
pipeline.run_pipeline()'''
|
|
|
|
|
|
|
|
|
|
from src.pipeline.inference import AssessmentInference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
inference = AssessmentInference(
|
|
|
|
|
company_id="testid",num_assessments=2
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
result = inference.run()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
print(result)
|
