synthesised categorical variable without encoding for SD2011 dataset

Author: KoomenErina

main.py

@@ -0,0 +1,30 @@
+from synthpop import Synthpop
+import pandas as pd
+import pyreadr
+
+
+def synSD2011():
+    df0 = pyreadr.read_r("SD2011.rda")['SD2011']
+    #pd.read_csv("bar_pass_prediction.csv")
+    print(df0)
+    df = df0[['age', 'unempdur', 'income', 'sex']]#df0[['sex', 'race1', 'ugpa', 'bar']]
+    #df.to_excel("inputData.xlsx")
+    dtype_map ={
+        "age":"float",
+        "unempdur":"float",
+        "income":"float",
+        "sex":"category"
+    }
+    #{'sex': 'float', 'race1': 'category', 'ugpa': 'float', 'bar': 'category'}
+    # for (k,v) in dtype_map.items():
+    #     if v == 'category':
+    #         df[k] = df[k].astype('category')
+
+    print(df.dtypes)
+    r = df.dtypes.keys()
+    spop = Synthpop()
+    spop.fit(df,dtype_map)
+
+    synth_df = spop.generate(len(df))
+
+    print(synth_df.head())