AlexWorldD / NetEmbs

Framework for Representation Learning on Financial Statement Networks
Apache License 2.0
1 stars 1 forks source link

loading data #2

Closed boersmamarcel closed 5 years ago

boersmamarcel commented 5 years ago

Please change the cell:

from NetEmbs.DataProcessing import *
YOUR_DATAFRAME = None
if YOUR_DATAFRAME is None:
    d = prepare_data(d)
else:
    d = prepare_data(rename_columns(YOUR_DATAFRAME), split=False)
d.head(20)

to

from NetEmbs.DataProcessing import *
df = pd.DataFrame.from_dict(journal_entries)
df.columns = ['ID', 'FA_Name', 'Debit', 'Credit']
YOUR_DATAFRAME = df
if YOUR_DATAFRAME is None:
    d = prepare_data(d)
else:
    d = prepare_data(YOUR_DATAFRAME, split=False)
d.head(20)

when I load the data frame it doesn't contain any names, therefore the replace function doesn't work as intended. The new code snippet fixes this.

AlexWorldD commented 5 years ago

done

boersmamarcel commented 5 years ago

Data loading is now smooth!:)