Skip to content

Latest commit

 

History

History
130 lines (94 loc) · 3.68 KB

README.md

File metadata and controls

130 lines (94 loc) · 3.68 KB

CI - Python Package codecov PyPI

Pandas2TensorBoard

Pandas2TensorBoard is a library for transforming the pandas DataFrame into the data fomrat of TensorBoard. Pandas2TensorBoard relies on torch utilities for the data transformation.

_

Installation

  • Regular installation via pip:

    pip install pandas2tensorboard
  • With modin backend for pandas

    pip install pandas2tensorboard[modin]
  • With Dask and omnisci support for pandas

    pip install pandas2tensorboard[backend]
  • With Dask and omnisci support for pandas and modin backend for pandas

    pip install pandas2tensorboard[all]

Usage

Currently the following types of export from pandas to tensorboard are supported:

  • pd.DataFramescalars
  • pd.DataFramescalars with timestamp
  • pd.DataFramescatter via hyperparameters

For initializing the pandas2tensorboard library, the current syntax of Torch's tensorboard.SummaryWriter is used.

Examples

  1. Exporting a regular pd.DataFrame to tensorboard by removing columns with str:

    import seaborn as sns
    
    from pandas2tensorboard import pandas2tensorboard as p2t
    
    pt = p2t.Pandas2TensorBoard()
    pt.regular_df(
        sns.load_dataset("planets"),
        label="planets",
        remove_nan=True,
        remove_str=True,
    )
    pt.close()
  2. Exporting a pd.DataFrame with time column to tensorboard:

    import seaborn as sns
    
    from pandas2tensorboard import pandas2tensorboard as p2t
    
    pt = p2t.Pandas2TensorBoard()
    pt.timeseries_df(
        sns.load_dataset("attention"),
        time="score",
        label="attention",
        remove_nan=True,
        remove_str=True,
        time_convert=True,
    )
    pt.close()

    The time column with name score is transformed into float with timestamp.

  3. Exporting a pd.DataFrame with hyperparameters to tensorboard:

    import seaborn as sns
    
    from pandas2tensorboard import pandas2tensorboard as p2t
    
    pt = p2t.Pandas2TensorBoard()
    pt.scatter_df(
        sns.load_dataset("anagrams"),
        x_axis="subidr",
        group="anagrams",
        remove_nan=True,
        remove_str=True,
    )
    pt.close()

    The x-axis corresponds to hparam_dict; the dataframe without column x_axis="subidr" corresponds to metric_dict.

Contributing

Please feel free to open an issue or create a pull request; see also contributing.

License

Check MIT