Adding risk adjusted benchmark mart

models/benchmarks/models.yml

@@ -0,0 +1,212 @@
+version: 2
+
+models:
+  - name: benchmarks__person_year
+    description: |
+        flat table for training ml models on 
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: person_year
+      materialized: table
+      tags: 
+        - benchmarks_train
+        - benchmarks
+    data_tests:
+      - dbt_utils.unique_combination_of_columns:
+          combination_of_columns:
+            - person_id
+            - payer
+            - "{{ '\"plan\"' if (target.type == 'fabric') else 'plan' }}"
+            - year_nbr      
+
+  - name: benchmarks__pivot_condition
+    description: |
+        flat table for training ml models on 
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: int_pivot_condition
+      materialized: table
+      tags: 
+        - benchmarks_train
+        - benchmarks
+
+
+  - name: benchmarks__predict_member_month
+    description: |
+        member_month grain table for data viz with expected values
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: predict_member_month
+      materialized: table
+      tags: 
+        - benchmarks_analysis
+        - benchmarks
+
+  - name: benchmarks__predict_inpatient
+    description: |
+        joining inpatient predicts to encounter table actuals
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: predict_inpatient
+      materialized: table
+      tags: 
+        - benchmarks_analysis
+        - benchmarks
+
+
+  - name: benchmarks__inpatient_input
+    description: |
+        unpivot for viz
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: benchmarks_inpatient_input
+      materialized: table
+      tags: 
+        - benchmarks_train
+        - benchmarks
+    data_tests:
+      - dbt_utils.unique_combination_of_columns:
+          combination_of_columns:
+            - encounter_id             
+
+  - name: benchmarks__unpivot_inpatient
+    description: |
+        unpivot for viz
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: benchmarks_unpivot_inpatient
+      materialized: table
+      tags: 
+        - benchmarks_analysis
+        - benchmarks
+
+  - name: benchmarks__unpivot_member_month
+    description: |
+        unpivot for viz
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: benchmarks_unpivot_member_month
+      materialized: table
+      tags: 
+        - benchmarks_analysis
+        - benchmarks
+
+
+  - name: benchmarks__pivot_hcc
+    description: |
+        hcc_person_year
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: int_pivot_hcc
+      materialized: table
+      tags: 
+        - benchmarks_train
+        - benchmarks
+
+  - name: benchmarks__pivot_cms_condition
+    description: |
+        hcc_person_year
+    config:
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+      alias: int_pivot_cms_condition
+      materialized: table
+      tags: 
+        - benchmarks_train
+        - benchmarks
+
+  # Staging models
+  - name: benchmarks__stg_core__condition
+    description: Staging model for core condition data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+
+
+  - name: benchmarks__stg_core__member_months
+    description: Staging model for core member months data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+
+  - name: benchmarks__stg_core__medical_claim
+    description: Staging model for core medical claim data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+
+  - name: benchmarks__stg_core__encounter
+    description: Staging model for core encounter data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+
+  - name: benchmarks__stg_core__patient
+    description: Staging model for core patient data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+
+  - name: benchmarks__stg_reference_data__calendar
+    description: Staging model for reference calendar data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}
+
+  - name: benchmarks__stg_cms_hcc__int_disease_factors
+    description: Staging model for CMS HCC disease factors data used in benchmarks
+    config:
+      materialized: view
+      tags: 
+        - benchmarks_train
+        - benchmarks      
+      schema: |
+        {%- if var('tuva_schema_prefix',None) != None -%}{{var('tuva_schema_prefix')}}_benchmarks
+        {%- else -%}benchmarks{%- endif -%}

models/benchmarks/post_predict/benchmarks__predict_inpatient.sql

@@ -0,0 +1,62 @@
+{{
+    config(
+        enabled = var('benchmarks_already_created', False) | as_bool
+    )
+}}
+
+WITH inpatient_pred AS (
+    SELECT 
+            p.encounter_id
+          , p.length_of_stay_pred
+          , p.readmission_pred
+          , p.discharge_location_pred
+          , p.discharge_pred_proba_expired
+          , p.discharge_pred_proba_home
+          , p.discharge_pred_proba_home_health
+          , p.discharge_pred_proba_hospice
+          , p.discharge_pred_proba_ipt_rehab
+          , p.discharge_pred_proba_other
+          , p.discharge_pred_proba_snf
+          , p.discharge_pred_proba_transfer_other_facility
+    FROM {{ var('predictions_inpatient') }} p 
+
+)
+
+,enrollment_flag as (
+    select encounter_id
+    ,max(enrollment_flag) as max_enrollment_flag
+    from {{ ref('benchmarks__stg_core__medical_claim') }} e
+    group by encounter_id 
+)
+
+
+select e.encounter_id
+, ce.data_source
+, ce.encounter_start_date
+, cal.first_day_of_month
+, ce.person_id
+, i.length_of_stay_pred as expected_los
+, e.length_of_stay as actual_los
+, e.readmission_numerator as actual_readmission
+, i.readmission_pred as expected_readmission
+, e.readmission_denominator
+, e.discharge_location as actual_discharge_location
+, i.discharge_location_pred as expected_discharge_location
+, ce.facility_id
+, ce.facility_name
+, case when ce.drg_code_type = 'ms-drg' then concat(ce.drg_code,' ',ce.drg_description) else null end as ms_drg_code
+, ce.paid_amount
+, i.discharge_pred_proba_expired
+, i.discharge_pred_proba_home
+, i.discharge_pred_proba_home_health
+, i.discharge_pred_proba_hospice
+, i.discharge_pred_proba_ipt_rehab
+, i.discharge_pred_proba_other
+, i.discharge_pred_proba_snf
+, i.discharge_pred_proba_transfer_other_facility
+, ef.max_enrollment_flag as enrolled_encounter_flag
+FROM {{ ref('benchmarks__inpatient_input') }}  e
+inner join inpatient_pred i on e.encounter_id = i.encounter_id
+inner join {{ ref('benchmarks__stg_core__encounter') }} ce on e.encounter_id = ce.encounter_id
+inner join {{ ref('benchmarks__stg_reference_data__calendar') }} cal on ce.encounter_start_date = cal.full_date
+left join enrollment_flag ef on e.encounter_id = ef.encounter_id