Commit ed49e1de authored by Alex Fout's avatar Alex Fout

updated stft method

parent fdc2040b
......@@ -2,7 +2,7 @@ import pandas as pd
from scipy import signal
def stft(session):
def stft(session, **kwargs):
:param session: session of eeg data with a raw instance variable containing a pandas data frame of channels
......@@ -13,4 +13,5 @@ def stft(session):
if not hasattr(session, "stft"):
session.stft = {}
for col in session.raw.columns:
session.stft[col] = signal.stft(session.raw[col])
session.stft[col] = signal.stft(session.raw[col], **kwargs)
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment