misleep 0.2.3__tar.gz → 0.2.4__tar.gz

{misleep-0.2.3 → misleep-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: misleep
-Version: 0.2.3
+Version: 0.2.4
 Summary: MiSleep: Mice Sleep EEG/EMG visualization, scoring and analysis.
 Home-page: https://github.com/BryanWang0702/MiSleep/
 Download-URL: https://github.com/BryanWang0702/MiSleep/

{misleep-0.2.3 → misleep-0.2.4}/misleep/analysis/auto_stage.py

@@ -6,6 +6,7 @@ from scipy import stats
 import antropy
 import joblib
 import copy
+import lightgbm
 
 
 def filter_power_line_noise(data, sf, noise_band='50-100-150'):
@@ -33,7 +34,7 @@ def split_window_data(data, sf, state, window_length=20, stride_length=5):
     """
 
     # If the data is shorter than the window length, remove it
-    if data.shape[0]/sf < 20:
+    if data.shape[0]/sf < window_length:
         return []
     
     window_data = []
@@ -45,15 +46,24 @@ def split_window_data(data, sf, state, window_length=20, stride_length=5):
 
     return window_data
 
-def delta_theta_ratio(data, sf):
+def delta_theta_ratio_theta(data, sf):
     """Calculate the delta/theta ratio from the original signal data
     Note, here the data is 20 seconds, we only need the ratio from the 5 seconds beginning
     """
     freq, t, Sxx = misleep.spectrogram(data, sf, window=1)
     band_second = np.where(t < 5)
     psd = np.sum(np.array([each[band_second] for each in Sxx]), axis=1)
-    band_power = misleep.band_power(psd, freq, bands=[[0.5, 4, 'delta'], [5, 9, 'theta']])
-    return band_power['delta'] / band_power['theta']
+    band_power = misleep.band_power(psd, freq, bands=[[0.5, 4, 'delta'], [5, 9, 'theta']], relative=True)
+    return band_power['delta'] / band_power['theta'], band_power['theta']
+
+def self_zscore(feature, quantile=0.95):
+    """Zscore the extracted features from the signal data, to exclude the abnormal signal, use 0.95 quantile
+    Return the quantile zscore feature data
+    """
+    upper_quantile = np.quantile(feature, quantile)
+    feature = [each if each < upper_quantile else upper_quantile for each in feature]
+    return (feature - np.mean(feature)) / np.std(feature)
+
 
 def get_data_features(data, sf, data_format='EEG'):
     """ Get data features, data format can be 'EEG' or 'EMG'
@@ -67,11 +77,8 @@ def get_data_features(data, sf, data_format='EEG'):
     # Time-domain features, both EEG and EMG
     # 1. standard deviation
     data_std = np.array([np.std(each[0][:int(5*sf)]) for each in data])
-    # Set those abnormal points to the upper quartile
-    data_std_upper_quantile = np.quantile(data_std, 0.95)
-    data_std = [each if each < data_std_upper_quantile else data_std_upper_quantile for each in data_std]
     # Z-score the std features
-    window_feature_df[f'{data_format}_std_zscore'] = (data_std - np.mean(data_std)) / np.std(data_std)
+    window_feature_df[f'{data_format}_std_zscore'] = self_zscore(data_std)
     
     # 2. Zero crossing rate for EEG and EMG
     zerocross_rate = [antropy.num_zerocross(each[0][:int(5*sf)]) / (5*sf) for each in data]
@@ -81,46 +88,60 @@ def get_data_features(data, sf, data_format='EEG'):
     hjorth = [antropy.hjorth_params(each[0][:int(5*sf)]) for each in data]
     hjorth_M = [each[0] for each in hjorth]
     hjorth_C = [each[1] for each in hjorth]
-    window_feature_df[f'{data_format}_Hjorth_M'] = (hjorth_M - np.mean(hjorth_M)) / np.std(hjorth_M)
-    window_feature_df[f'{data_format}_Hjorth_C'] = (hjorth_C - np.mean(hjorth_C)) / np.std(hjorth_C)
+    window_feature_df[f'{data_format}_Hjorth_M'] = self_zscore(hjorth_M)
+    window_feature_df[f'{data_format}_Hjorth_C'] = self_zscore(hjorth_C)
 
     # 4. Permutation entropy
     perm_entropy= [antropy.perm_entropy(each[0][:int(5*sf)]) for each in data]
-    window_feature_df[f'{data_format}_perm_entropy'] = (perm_entropy - np.mean(perm_entropy)) / np.std(perm_entropy) 
+    window_feature_df[f'{data_format}_perm_entropy'] = self_zscore(perm_entropy)
 
     # Some features only with EEG
     if data_format.startswith('EEG'):
         # 1. Skewness and kurtosis for EEG signal(s)
         data_skewness = np.array([stats.skew(each[0][:int(5*sf)]) for each in data])
         data_kurtosis = np.array([stats.kurtosis(each[0][:int(5*sf)]) for each in data])
-        window_feature_df[f'{data_format}_skewness_zscore'] = (data_skewness - np.mean(data_skewness)) / np.std(data_skewness)
-        window_feature_df[f'{data_format}_kurtosis_zscore'] = (data_kurtosis - np.mean(data_kurtosis)) / np.std(data_kurtosis)
+        window_feature_df[f'{data_format}_skewness_zscore'] = self_zscore(data_skewness)
+        window_feature_df[f'{data_format}_kurtosis_zscore'] = self_zscore(data_kurtosis)
         # Frequency-domain features
         # 1. delta/theta
-        delta_theta = [delta_theta_ratio(each[0], sf) for each in data]
-        window_feature_df[f'{data_format}_delta_theta'] = (delta_theta - np.mean(delta_theta)) / np.std(delta_theta) 
+        delta_theta = [delta_theta_ratio_theta(each[0], sf) for each in data]
+        delta_theta_ratio = [each[0] for each in delta_theta]
+        theta = [each[1] for each in delta_theta]
+        window_feature_df[f'{data_format}_delta_theta_ratio'] = self_zscore(delta_theta_ratio)
+        window_feature_df[f'{data_format}_theta'] = self_zscore(theta)
 
     return window_feature_df
 
-def result_constraints(pred_label):
+def result_constraints(pred_prob):
     """
     Once finished the prediction, add some constraints to make the result more smooth
     1. No REM after Wake, set to NREM
     2. All one epoch between two same state will be set to the around state
     """
 
-    pred_label = copy.deepcopy(pred_label)
-    for idx in range(1, len(pred_label)-1):
-        label_ = pred_label[idx]
-        if label_ == 3 and pred_label[idx+1] == 2:  # REM after Wake
-            pred_label[idx+1] = 1
-        if pred_label[idx-1] ==pred_label[idx+1]:  # Same state previous and after
-            pred_label[idx] = pred_label[idx-1]
+    """
+    Once finished the prediction, add some constraints to make the result more smooth
+    1. REM threshold is lower, set to 0.15
+    2. No REM after Wake, set to NREM
+    3. All one epoch between two same state will be set to the around state
+    """
+
+    pred_prob = copy.deepcopy(pred_prob)
+    pred_label = [each+1 for each in np.argmax(pred_prob, axis=1)]
+    pred_label = [2 if each[1] > 0.1 else pred_label[idx] for idx, each in enumerate(pred_prob)]  # REM threshold
+        
+    # for idx in range(1, len(pred_label)-1):
+    #     label_ = pred_label[idx]
+
+    #     if label_ == 3 and pred_label[idx+1] == 2:  # REM after Wake
+    #         pred_label[idx+1] = 1
+    #     if pred_label[idx-1] ==pred_label[idx+1]:  # Same state previous and after
+    #         pred_label[idx] = pred_label[idx-1]
     
     return pred_label
 
 
-def auto_stage_gbm(EEG, EMG, sf, EEG_channel='F'):
+def auto_stage_gbm(EEG, EMG, label, sf, EEG_channel='F', mouse_age='adult'):
     """
     Auto stage with lightgbm method
     
@@ -134,12 +155,16 @@ def auto_stage_gbm(EEG, EMG, sf, EEG_channel='F'):
         Sampling frequency of the EEG and EMG data, should be the same.
     EEG_channel : string
         Specify the channel of EEG, frontal or parietal. Options: {'F', 'P'}. Default is 'F' for frontal.
+    mouse_age : string
+        Specify which age model to use. {'adult', 'ado', 'P30'}, default is 'adult'.
+        Here the adult is > P56, ado is P30~P56, P30 is less than P30.
 
     Return
     ------
     pred_label : list
         Predicted labels for every second. May less than the data length for about 15 seconds.
     """ 
+
     EEG = split_window_data(EEG, sf, state=4)  # All set the initial state '4'
     EMG = split_window_data(EMG, sf, state=4)
 
@@ -147,12 +172,9 @@ def auto_stage_gbm(EEG, EMG, sf, EEG_channel='F'):
     window_feature_df = pd.concat([get_data_features(EEG, sf, data_format='EEG'), get_data_features(EMG, sf, data_format='EMG')], axis=1)
     window_feature_df = window_feature_df.filter(like='E')
 
-    if EEG_channel == 'F':
-        gbm_model = joblib.load(r'./misleep/analysis/auto_stage_model/EEG_F_lightgbm_20241221.pkl')
-    if EEG_channel == 'P':
-        gbm_model = joblib.load(r'./misleep/analysis/auto_stage_model/EEG_P_lightgbm_20241221.pkl')
+    gbm_model = joblib.load(f'./misleep/analysis/auto_stage_model/{mouse_age}_EEG_{EEG_channel}_lightgbm.pkl')
 
-    pred_label = gbm_model.predict(window_feature_df, num_iteration=gbm_model.best_iteration_)
-    pred_label = result_constraints(pred_label)
+    pred_prob = gbm_model.predict_proba(window_feature_df, num_iteration=gbm_model.best_iteration_)
+    pred_label = result_constraints(pred_prob)
     pred_label = [item for each in pred_label for item in [each]*5]
     return pred_label

{misleep-0.2.3 → misleep-0.2.4}/misleep/config.ini

@@ -1,7 +1,7 @@
 [gui]
-version = v0.2.3
-updatetime = 2024/12/23
-marker = ['long REM', 'first REM', 'WindEEG', 'W-R ', 'maker']
+version = v0.2.4
+updatetime = 2025/1/16
+marker = ['good', 'first REM', 'WindEEG', 'W-R ', 'maker']
 startend = ['burst-supression', 'REM', 'Wake', 'Spindle', 'SWA', 'start end label', 'start end label']
 statemap = {"1": "NREM", "2": "REM", "3": "Wake", "4": "INIT", "5": "IS", "6": "MicroArousal"}
 statecolor = {"1": "orange", "2": "skyblue", "3": "red", "4": "white", "5": "green", "6": "pink"}
@@ -10,4 +10,3 @@ statecolorbgalpha = 0.1
 markerlinecolor = "red"
 startendlinecolor = "blue"
 openpath = E:
-

{misleep-0.2.3 → misleep-0.2.4}/misleep/gui/dialog.py

@@ -220,7 +220,7 @@ class transferResult_dialog(QDialog, Ui_TransferResultDialog):
         df, analyse_df, start_end_df, marker_df = transfer_result(mianno=mianno, ac_time=ac_time)        
         
         fd, _ = QFileDialog.getSaveFileName(self, "Save transfered result",
-                                                f"{config['gui']['openpath'].split('/')[0]}/transfer_result.xlsx", 
+                                                f"{config['gui']['openpath']}", 
                                                 "*.xlsx;;")
         if fd == '':
             return
@@ -901,20 +901,26 @@ class AutoStageDialog(QDialog, Ui_AutoStageDialog):
         self.EMGchannelCombox.addItems(channels)
         self.EMGchannelCombox.setCurrentIndex(1)
 
-    def auto_stage(self, midata):
+    def auto_stage(self, midata, mianno):
         """Auto stage with misleep data"""
 
         EEG_channel_idx = self.EEGChannelCombox.currentIndex()
         EMG_channel_idx = self.EMGchannelCombox.currentIndex()
         EEG = deepcopy(midata.signals[EEG_channel_idx])
         EMG = deepcopy(midata.signals[EMG_channel_idx])
+        label = deepcopy(mianno._sleep_state)
         sf = deepcopy(midata.sf[EEG_channel_idx])
 
         EEG_site = ['P', 'F'][self.EEGSiteCombox.currentIndex()]
+        mouse_age = ['adult', 'ado', 'P30'][self.AgeCombox.currentIndex()]
 
-        pred_label = auto_stage_gbm(EEG=EEG, EMG=EMG, sf=sf, EEG_channel=EEG_site)
+        pred_label = auto_stage_gbm(EEG=EEG, EMG=EMG, label=label, sf=sf, EEG_channel=EEG_site, mouse_age=mouse_age)
+        if self.SaveAnnoCheckbox.isChecked():
+            save_anno = True
+        else: 
+            save_anno = False
 
-        return pred_label
+        return pred_label, save_anno
 
 
     def okEvent(self):

{misleep-0.2.3 → misleep-0.2.4}/misleep/gui/main_window.py

@@ -1624,11 +1624,13 @@ class main_window(QMainWindow, Ui_MiSleep):
             self.auto_stage_dialog.exec()
             if self.auto_stage_dialog.closed:
                 return
-            auto_stage_lst = self.auto_stage_dialog.auto_stage(self.midata)
+            auto_stage_lst, save_anno = self.auto_stage_dialog.auto_stage(self.midata, self.mianno)
             
             # Not sure which one is shorter, so get the minimum one
             self.mianno._sleep_state[:min(len(self.mianno._sleep_state),len(auto_stage_lst))] = auto_stage_lst[:min(len(self.mianno._sleep_state),len(auto_stage_lst))] 
 
+            if save_anno:
+                self.save_anno()
             self.is_saved = False
             self.AnnotationPathLabel.setText('*Annotation path:')
             self.plot_signals()
@@ -1638,8 +1640,11 @@ class main_window(QMainWindow, Ui_MiSleep):
             QMessageBox.about(self, "Error", "Auto stage ERROR")
             return
 
-    def save_anno(self):
-        """Save annotation into file"""
+    def save_anno(self, just_save=False):
+        """Save annotation into file
+        Just save: bool
+            For when I only want to save a file but not cover the current labels
+        """
         if self.anno_path == "":
             anno_path, _ = QFileDialog.getOpenFileName(
             self, "Select annotation file", 
@@ -1649,15 +1654,15 @@ class main_window(QMainWindow, Ui_MiSleep):
 
             if anno_path == "":
                 return
-            self.anno_path = anno_path
-            self.AnnoPathEdit.setText(self.anno_path)
+            if not just_save:
+                self.anno_path = anno_path
+                self.AnnoPathEdit.setText(self.anno_path)
 
         save_thread = SaveThread(file=[self.mianno, self.midata], 
                                  file_path=self.anno_path)
         saved = save_thread.save_anno()
-        if saved:
+        if saved and not just_save:
             self.is_saved = True
-            
             self.AnnotationPathLabel.setText('Annotation path:')
         save_thread.quit()
 

{misleep-0.2.3 → misleep-0.2.4}/misleep/gui/uis/auto_stage_dialog_ui.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-# Form implementation generated from reading ui file '.\auto_stage_dialog.ui'
+# Form implementation generated from reading ui file './misleep/gui/uis/auto_stage_dialog.ui'
 #
 # Created by: PyQt5 UI code generator 5.15.9
 #
@@ -14,7 +14,7 @@ from PyQt5 import QtCore, QtGui, QtWidgets
 class Ui_AutoStageDialog(object):
     def setupUi(self, AutoStageDialog):
         AutoStageDialog.setObjectName("AutoStageDialog")
-        AutoStageDialog.resize(347, 265)
+        AutoStageDialog.resize(408, 342)
         icon = QtGui.QIcon()
         icon.addPixmap(QtGui.QPixmap(":/logo/logo.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
         AutoStageDialog.setWindowIcon(icon)
@@ -30,26 +30,38 @@ class Ui_AutoStageDialog(object):
         self.groupBox.setObjectName("groupBox")
         self.gridLayout_2 = QtWidgets.QGridLayout(self.groupBox)
         self.gridLayout_2.setObjectName("gridLayout_2")
-        self.label_2 = QtWidgets.QLabel(self.groupBox)
-        self.label_2.setObjectName("label_2")
-        self.gridLayout_2.addWidget(self.label_2, 2, 0, 1, 1)
+        self.AgeCombox = QtWidgets.QComboBox(self.groupBox)
+        self.AgeCombox.setObjectName("AgeCombox")
+        self.AgeCombox.addItem("")
+        self.AgeCombox.addItem("")
+        self.AgeCombox.addItem("")
+        self.gridLayout_2.addWidget(self.AgeCombox, 3, 1, 1, 2)
         self.label = QtWidgets.QLabel(self.groupBox)
         self.label.setObjectName("label")
         self.gridLayout_2.addWidget(self.label, 0, 0, 1, 1)
-        self.label_8 = QtWidgets.QLabel(self.groupBox)
-        self.label_8.setObjectName("label_8")
-        self.gridLayout_2.addWidget(self.label_8, 1, 0, 1, 1)
+        self.label_3 = QtWidgets.QLabel(self.groupBox)
+        self.label_3.setObjectName("label_3")
+        self.gridLayout_2.addWidget(self.label_3, 3, 0, 1, 1)
         self.EMGchannelCombox = QtWidgets.QComboBox(self.groupBox)
         self.EMGchannelCombox.setObjectName("EMGchannelCombox")
         self.gridLayout_2.addWidget(self.EMGchannelCombox, 1, 1, 1, 2)
         self.EEGChannelCombox = QtWidgets.QComboBox(self.groupBox)
         self.EEGChannelCombox.setObjectName("EEGChannelCombox")
         self.gridLayout_2.addWidget(self.EEGChannelCombox, 0, 1, 1, 2)
+        self.label_2 = QtWidgets.QLabel(self.groupBox)
+        self.label_2.setObjectName("label_2")
+        self.gridLayout_2.addWidget(self.label_2, 2, 0, 1, 1)
+        self.label_8 = QtWidgets.QLabel(self.groupBox)
+        self.label_8.setObjectName("label_8")
+        self.gridLayout_2.addWidget(self.label_8, 1, 0, 1, 1)
         self.EEGSiteCombox = QtWidgets.QComboBox(self.groupBox)
         self.EEGSiteCombox.setObjectName("EEGSiteCombox")
         self.EEGSiteCombox.addItem("")
         self.EEGSiteCombox.addItem("")
         self.gridLayout_2.addWidget(self.EEGSiteCombox, 2, 1, 1, 2)
+        self.SaveAnnoCheckbox = QtWidgets.QCheckBox(self.groupBox)
+        self.SaveAnnoCheckbox.setObjectName("SaveAnnoCheckbox")
+        self.gridLayout_2.addWidget(self.SaveAnnoCheckbox, 4, 0, 1, 3)
         self.gridLayout.addWidget(self.groupBox, 0, 0, 1, 2)
 
         self.retranslateUi(AutoStageDialog)
@@ -61,9 +73,14 @@ class Ui_AutoStageDialog(object):
         self.OKBt.setText(_translate("AutoStageDialog", "Ok"))
         self.CancelBt.setText(_translate("AutoStageDialog", "Cancel"))
         self.groupBox.setTitle(_translate("AutoStageDialog", "Auto Stage options"))
-        self.label_2.setText(_translate("AutoStageDialog", "EEG site:"))
+        self.AgeCombox.setItemText(0, _translate("AutoStageDialog", "> P56"))
+        self.AgeCombox.setItemText(1, _translate("AutoStageDialog", "P30 - P56"))
+        self.AgeCombox.setItemText(2, _translate("AutoStageDialog", "< P30"))
         self.label.setText(_translate("AutoStageDialog", "EEG channel:"))
+        self.label_3.setText(_translate("AutoStageDialog", "Mouse age:"))
+        self.label_2.setText(_translate("AutoStageDialog", "EEG site:"))
         self.label_8.setText(_translate("AutoStageDialog", "EMG channel:"))
         self.EEGSiteCombox.setItemText(0, _translate("AutoStageDialog", "Parietal"))
         self.EEGSiteCombox.setItemText(1, _translate("AutoStageDialog", "Frontal"))
+        self.SaveAnnoCheckbox.setText(_translate("AutoStageDialog", "Cover current label"))
 from misleep.gui.resources import misleep

{misleep-0.2.3 → misleep-0.2.4}/misleep/io/annotation_io.py

@@ -135,6 +135,34 @@ def transfer_result(mianno, ac_time):
     ] = analyse_df[
         ['NREM_duration', 'NREM_bout', 'REM_duration', 'REM_bout', 'WAKE_duration',
         'WAKE_bout', 'INIT_duration', 'INIT_bout']].astype(int)
+
+
+    # gather the 12-h data to output the light phase and dark phase, set the first hour as ZT0 by default
+
+    try:
+        phase_data = pd.DataFrame()
+        phase_data['date_time'] = ['ZT0-ZT12', 'ZT12-ZT24']
+        phase_data['NREM_duration'] = [analyse_df['NREM_duration'].iloc[:12].sum(), analyse_df['NREM_duration'].iloc[12:24].sum()]
+        phase_data['NREM_bout'] = [analyse_df['NREM_bout'].iloc[:12].sum(), analyse_df['NREM_bout'].iloc[12:24].sum()]
+        phase_data['NREM_ave'] = phase_data.apply(lambda x: x['NREM_duration']/x['NREM_bout'] if x['NREM_bout']!=0 else 0, axis=1)
+        phase_data['NREM_percentage'] = phase_data.apply(lambda x: x['NREM_duration'] / 3600*12, axis=1)
+        phase_data['REM_duration'] = [analyse_df['REM_duration'].iloc[:12].sum(), analyse_df['REM_duration'].iloc[12:24].sum()]
+        phase_data['REM_bout'] = [analyse_df['REM_bout'].iloc[:12].sum(), analyse_df['REM_bout'].iloc[12:24].sum()]
+        phase_data['REM_ave'] = phase_data.apply(lambda x: x['REM_duration']/x['REM_bout'] if x['REM_bout']!=0 else 0, axis=1)
+        phase_data['REM_percentage'] = phase_data.apply(lambda x: x['REM_duration'] / 3600*12, axis=1)
+        phase_data['WAKE_duration'] = [analyse_df['WAKE_duration'].iloc[:12].sum(), analyse_df['WAKE_duration'].iloc[12:24].sum()]
+        phase_data['WAKE_bout'] = [analyse_df['WAKE_bout'].iloc[:12].sum(), analyse_df['WAKE_bout'].iloc[12:24].sum()]
+        phase_data['WAKE_ave'] = phase_data.apply(lambda x: x['WAKE_duration']/x['WAKE_bout'] if x['WAKE_bout']!=0 else 0, axis=1)
+        phase_data['WAKE_percentage'] = phase_data.apply(lambda x: x['WAKE_duration'] / 3600*12, axis=1)
+        phase_data['INIT_duration'] = [analyse_df['INIT_duration'].iloc[:12].sum(), analyse_df['INIT_duration'].iloc[12:24].sum()]
+        phase_data['INIT_bout'] = [analyse_df['INIT_bout'].iloc[:12].sum(), analyse_df['INIT_bout'].iloc[12:24].sum()]
+        phase_data['INIT_ave'] = phase_data.apply(lambda x: x['INIT_duration']/x['INIT_bout'] if x['INIT_bout']!=0 else 0, axis=1)
+        phase_data['INIT_percentage'] = phase_data.apply(lambda x: x['INIT_duration'] / 3600*12, axis=1)
+
+        analyse_df = pd.concat([analyse_df, phase_data])
+        analyse_df.reset_index(inplace=True)
+    except Exception as e:
+        pass
     
     start_end_df = pd.DataFrame(start_end_label, 
                                 columns=['start_time', 'start_time_sec', 

{misleep-0.2.3 → misleep-0.2.4}/misleep.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: misleep
-Version: 0.2.3
+Version: 0.2.4
 Summary: MiSleep: Mice Sleep EEG/EMG visualization, scoring and analysis.
 Home-page: https://github.com/BryanWang0702/MiSleep/
 Download-URL: https://github.com/BryanWang0702/MiSleep/

{misleep-0.2.3 → misleep-0.2.4}/misleep.egg-info/SOURCES.txt

@@ -11,10 +11,13 @@ misleep.egg-info/requires.txt
 misleep.egg-info/top_level.txt
 misleep/analysis/__init__.py
 misleep/analysis/auto_stage.py
-misleep/analysis/classification.py
 misleep/analysis/detection.py
-misleep/analysis/auto_stage_model/EEG_F_lightgbm_20241221.pkl
-misleep/analysis/auto_stage_model/EEG_P_lightgbm_20241221.pkl
+misleep/analysis/auto_stage_model/P30_EEG_F_lightgbm.pkl
+misleep/analysis/auto_stage_model/P30_EEG_P_lightgbm.pkl
+misleep/analysis/auto_stage_model/ado_EEG_F_lightgbm.pkl
+misleep/analysis/auto_stage_model/ado_EEG_P_lightgbm.pkl
+misleep/analysis/auto_stage_model/adult_EEG_F_lightgbm.pkl
+misleep/analysis/auto_stage_model/adult_EEG_P_lightgbm.pkl
 misleep/gui/__init__.py
 misleep/gui/about.py
 misleep/gui/dialog.py

{misleep-0.2.3 → misleep-0.2.4}/setup.py

@@ -11,7 +11,7 @@ MAINTAINER_EMAIL = "swang@gmail.com"
 URL = "https://github.com/BryanWang0702/MiSleep/"
 LICENSE = "BSD (3-clause)"
 DOWNLOAD_URL = "https://github.com/BryanWang0702/MiSleep/"
-VERSION = "0.2.3"
+VERSION = "0.2.4"
 
 INSTALL_REQUIRES = [
     "numpy>=1.18.1",
@@ -51,8 +51,12 @@ DATA_FILES = [
             "misleep/gui/resources/misleep.qrc",
             "misleep/gui/resources/logo.png",
             "misleep/gui/resources/entire_logo.png",
-            "misleep/analysis/auto_stage_model/EEG_F_lightgbm_20241221.pkl",
-            "misleep/analysis/auto_stage_model/EEG_P_lightgbm_20241221.pkl"
+            "misleep/analysis/auto_stage_model/adult_EEG_F_lightgbm.pkl",
+            "misleep/analysis/auto_stage_model/adult_EEG_P_lightgbm.pkl",
+            "misleep/analysis/auto_stage_model/ado_EEG_F_lightgbm.pkl",
+            "misleep/analysis/auto_stage_model/ado_EEG_P_lightgbm.pkl",
+            "misleep/analysis/auto_stage_model/P30_EEG_F_lightgbm.pkl",
+            "misleep/analysis/auto_stage_model/P30_EEG_P_lightgbm.pkl"
         ],
     )
 ]