Commit 48590f7f authored by Ibrahim's avatar Ibrahim
Browse files

Import error for tslearn fixed

parent 76c03b44
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -2,3 +2,4 @@
__pycache__
*.ipynb_checkopoints
.vscode
*.egg-info
+12 −0
Original line number Diff line number Diff line
@@ -3,6 +3,18 @@ import pandas as pd


def get_tensorboard_scalar_frame(path: str) -> pd.DataFrame:
    """
    Get a pandas dataframe containing scalar values logged by tensorboard.

    Parameters
    ----------
    path : str
        Path to directory containing tensorboard log files.

    Returns
    -------
    pd.DataFrame
    """
    ea = event_accumulator.EventAccumulator(
        path=path,
        size_guidance={event_accumulator.SCALARS: 0}
+2 −2
Original line number Diff line number Diff line
@@ -7,8 +7,8 @@ from typing import List
import numpy as np
import pandas as pd
from sklearn.preprocessing import MinMaxScaler
from tslearn.metrics import dtw
from tslearn.utils import to_time_series
# from tslearn.metrics import dtw
# from tslearn.utils import to_time_series



+0 −1
Original line number Diff line number Diff line
@@ -12,7 +12,6 @@ python_requires = >= 3.7
install_requires =
    numpy
    scipy
    numba
    matplotlib
    gym
    torch