ElixhauserGenerator.h

Go to the documentation of this file.

#pragma once
#include "FeatureGenerator.h"
 
class ElixhauserGenerator : public FeatureGenerator {
private:
    vector<string> types = { "CONGESTIVE_HEART_FAILURE","CARDIAC_ARRHYTHMIAS","VALVULAR_DISEASE","PULMONARY_CIRCULATION","PERIPHERAL_VASCULAR","HYPERTENSION","PARALYSIS","OTHER_NEUROLOGICAL",
        "CHRONIC_PULMONARY","DIABETES_UNCOMPLICATED","DIABETES_COMPLICATED","HYPOTHYROIDISM","RENAL_FAILURE","LIVER_DISEASE","PEPTIC_ULCER","AIDS","LYMPHOMA","METASTATIC_CANCER","SOLID_TUMOR",
        "RHEUMATOID_ARTHRITIS","COAGULOPATHY","OBESITY","WEIGHT_LOSS","FLUID_ELECTROLYTE","BLOOD_LOSS_ANEMIA","DEFICIENCY_ANEMIAS","ALCOHOL_ABUSE","DRUG_ABUSE","PSYCHOSES","DEPRESSION" };
    map<string, vector<int>> _weights = { {"ahrq",{9,0,0,6,3,-1,5,5,3,0,-3,0,6,4,0,0,6,14,7,0,11,-5,9,11,-3,-2,-1,-7,-5,-5}},{"vanwarlaven",{7,5,-1,4,2,0,7,6,3,0,0,0,5,11,0,0,9,12,4,0,3,-4,6,5,-2,-2,0,-7,0,-3}} };
    vector<vector<string>> drgSets, diagSets;
    vector<vector<unsigned char>> drgLuts,diagLuts;
 
    void parseSets(string& init_string, vector<vector<string>>& sets);
    void initSets();
 
    // Helper
    vector<int> weights;
 
public:
    // Feature Descrption
    string drgSignalName = "DRG_IP", diagSignalName = "DIAGNOSIS_IP";
    int drgSignalId, diagSignalId;
    string type; // ahrq/vanwarlaven
 
    // parameters (should be serialized)
    int win_from = 0;
    int win_to = 360000;
    int time_unit_win = MedTime::Undefined;         
    int drg_time_unit_sig = MedTime::Undefined;     
    int diag_time_unit_sig = MedTime::Undefined;        
 
    // Constructor/Destructor
    ElixhauserGenerator() : FeatureGenerator() {
        generator_type = FTR_GEN_ELIXHAUSER; req_signals = { drgSignalName,diagSignalName }; initSets();
    }
 
    ~ElixhauserGenerator() {};
 
    virtual int init(map<string, string>& mapper);
 
    // Name
    void set_names() { if (names.empty()) names.push_back("FTR_" + int_to_string_digits(serial_id, 6) + ".Elixhauser"); tags.push_back("Elixhauser"); }
 
    // Copy
    virtual void copy(FeatureGenerator *generator) { *this = *(dynamic_cast<ElixhauserGenerator *>(generator)); }
 
    void init_tables(MedDictionarySections& dict);
 
    // Signal Ids
    void set_signal_ids(MedSignals& sigs) { drgSignalId = sigs.sid(drgSignalName); diagSignalId = sigs.sid(diagSignalName); }
 
    // Learn a generator
    int _learn(MedPidRepository& rep, const MedSamples& samples, vector<RepProcessor *> processors);
 
    // generate a new feature
    int _generate(PidDynamicRec& rec, MedFeatures& features, int index, int num, vector<float *> &_p_data);
 
    // Serialization
    ADD_CLASS_NAME(ElixhauserGenerator)
    ADD_SERIALIZATION_FUNCS(generator_type, types, type, drgSignalName, diagSignalName, drgSets, diagSets, req_signals);
};