aou/data-maker
aou
/
data-maker
1
1
Fork 0
Code Issues 4 Pull Requests Packages Projects 1 Releases Wiki Activity

bug fix , generator

This commit is contained in:
Steve Nyemba 2020-02-20 22:37:25 -06:00
parent 74c1f9d511
commit cd88a9660a
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
data/gan.py
View File

@ -581,6 +581,7 @@ class Predict(GNet):
# #
# In case we are dealing with actual values like diagnosis codes we can perform # In case we are dealing with actual values like diagnosis codes we can perform
# #
df = found[np.random.choice(np.arange(len(found)),1)[0]] df = found[np.random.choice(np.arange(len(found)),1)[0]]
columns = self.ATTRIBUTES['synthetic'] if isinstance(self.ATTRIBUTES['synthetic'],list)else [self.ATTRIBUTES['synthetic']] columns = self.ATTRIBUTES['synthetic'] if isinstance(self.ATTRIBUTES['synthetic'],list)else [self.ATTRIBUTES['synthetic']]

2
data/maker/__init__.py
View File

@ -38,7 +38,7 @@ def train (**args) :
# args['label'] = handler.Export(df[[column_id]]) # args['label'] = handler.Export(df[[column_id]])
# args['label'] = np.ones(df.shape[0]).reshape(df.shape[0],1) # args['label'] = np.ones(df.shape[0]).reshape(df.shape[0],1)
for col in column : for col in column :
args['real'] = pd.get_dummies(df[col]).astype(np.float32).values args['real'] = pd.get_dummies(df[col].fillna('')).astype(np.float32).values
# args['real'] = handler.Export(df[[col]]) # args['real'] = handler.Export(df[[col]])
args['column'] = col args['column'] = col
args['context'] = col args['context'] = col