assessment predictions pipelines added

2024-09-12 21:36:02 +00:00
parent 24823432b3
commit 45bc62c745
10 changed files with 1583 additions and 11 deletions
@@ -1,7 +1,20 @@
 # 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
+'''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

 pipeline = CompanyModelPipeline(company_ids=company_ids, input_base_path=input_base_path)
-pipeline.run_pipeline()
+pipeline.run_pipeline()'''
+
+from src.pipeline.inference import AssessmentInference
+
+
+
+inference = AssessmentInference(
+    company_id="testid",num_assessments=2
+)
+
+result = inference.run()
+
+
+print(result)