Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how should i do when i use my data using dataframe not loaded csv file? #2

Open
heemokyim opened this issue Aug 20, 2019 · 1 comment

Comments

@heemokyim
Copy link

heemokyim commented Aug 20, 2019

for example,

from entity_embeddings import Config, Embedder, TargetType
from entity_embeddings.util import visualization_utils

df=pd.read_csv('test_file.csv')

config = Config.make_default_config(df,
target_name='churn',
target_type=TargetType.BINARY_CLASSIFICATION,
train_ratio=0.8,
epochs=10,
verbose=True,
artifacts_path='artifacts')

embedder = Embedder(config)
embedder.perform_embedding()

@rodrigobressan
Copy link
Owner

Hello @heemokyim, thanks for the question. At this moment we're only accepting the path of the csv file.

If you need the DataFrame to be used as parameter, I can look into implementing it in the following weeks, or if you need it sooner, feel free to submit a pull request with the implementation. Certainly it would be welcome to the project.

In short, you just need to add a new initializer into the Config class (config.py file) with the pandas DataFrame instead of the csv file path.

If you have any further questions, please do not hesitate to ask.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants