Skip to content

Commit

Permalink
change sample_
Browse files Browse the repository at this point in the history
  • Loading branch information
rodrigosnader committed Jul 1, 2022
1 parent 7058b18 commit d8e2c3c
Show file tree
Hide file tree
Showing 13 changed files with 150,822 additions and 193,962 deletions.
1,985 changes: 1,985 additions & 0 deletions data/cnn_dataset/Processed_DJI.csv

Large diffs are not rendered by default.

1,985 changes: 1,985 additions & 0 deletions data/cnn_dataset/Processed_NASDAQ.csv

Large diffs are not rendered by default.

1,985 changes: 1,985 additions & 0 deletions data/cnn_dataset/Processed_NYSE.csv

Large diffs are not rendered by default.

1,985 changes: 1,985 additions & 0 deletions data/cnn_dataset/Processed_RUSSELL.csv

Large diffs are not rendered by default.

1,985 changes: 1,985 additions & 0 deletions data/cnn_dataset/Processed_SP.csv

Large diffs are not rendered by default.

88,833 changes: 0 additions & 88,833 deletions notebooks/msft_analysis.1.ipynb

This file was deleted.

140,182 changes: 70,091 additions & 70,091 deletions notebooks/tf_processed_data.csv

Large diffs are not rendered by default.

696 changes: 106 additions & 590 deletions notebooks/tf_time_series.ipynb

Large diffs are not rendered by default.

1,422 changes: 0 additions & 1,422 deletions notebooks/tf_tutorial.ipynb

This file was deleted.

33 changes: 17 additions & 16 deletions src/wavy/panel.py
Original file line number Diff line number Diff line change
Expand Up @@ -886,31 +886,32 @@ def tail(self, n: int = 5):
"""
return self[-n:]

def sample(self, samples: int = 5, how: str = "first"):
def sample_(self, samples: int = 5, how: str = "random"):
"""
Sample panel returning a subset of frames.
Args:
samples (int): Number of samples to keep
how (str): Resempling how, 'first', 'last', 'spaced' or 'random'
how (str): Sampling method, 'spaced' or 'random'
Returns:
``Panel``: Result of sample function.
"""

if how == "first":
return self[:samples]
elif how == "last":
return self[-samples:]
elif how == "random":
indexes = np.random.choice(len(self.frames), samples, replace=False)
indexes = sorted(indexes)
return self[indexes]
elif how == "spaced":
indexes = np.linspace(
0, len(self.frames), samples, dtype=int, endpoint=False
)
return self[indexes]
# ! Needs to check if panel has train, test and val split first
# if panel.is_split():
# do something else
# ! Should sample train, val and test separately

# ! Better not to use below and focus on ids
# if how == "random":
# indexes = np.random.choice(len(self.frames), samples, replace=False)
# indexes = sorted(indexes)
# return self[indexes]
# elif how == "spaced":
# indexes = np.linspace(
# 0, len(self.frames), samples, dtype=int, endpoint=False
# )
# return self[indexes]

@property
def train(self):
Expand Down
403 changes: 403 additions & 0 deletions tutorials/mlm_cnn.ipynb

Large diffs are not rendered by default.

Loading

0 comments on commit d8e2c3c

Please sign in to comment.