Fix HR intraday bug: minutesonZONE features are 0

pull/114/head
Meng Li 2021-02-01 13:57:12 -05:00
parent aefc794274
commit f83c5a585e
1 changed files with 3 additions and 2 deletions

View File

@ -55,9 +55,10 @@ def extractHRFeaturesFromIntradayData(heartrate_intraday_data, features, time_se
# get number of minutes in each heart rate zone # get number of minutes in each heart rate zone
for feature_name in list(set(["minutesonoutofrangezone", "minutesonfatburnzone", "minutesoncardiozone", "minutesonpeakzone"]) & set(features)): for feature_name in list(set(["minutesonoutofrangezone", "minutesonfatburnzone", "minutesoncardiozone", "minutesonpeakzone"]) & set(features)):
heartrate_zone = heartrate_intraday_data[heartrate_intraday_data["heartrate_zone"] == feature_name[17:-4]] heartrate_zone = heartrate_intraday_data[heartrate_intraday_data["heartrate_zone"] == feature_name[9:-4]]
heartrate_intraday_features[feature_name] = heartrate_zone.groupby(["local_segment"])["device_id"].count() / num_rows_per_minute heartrate_intraday_features[feature_name] = heartrate_zone.groupby(["local_segment"])["device_id"].count() / num_rows_per_minute
heartrate_intraday_features.fillna(value={feature_name: 0}, inplace=True) heartrate_intraday_features.fillna(value={feature_name: 0}, inplace=True)
heartrate_intraday_features.reset_index(inplace=True) heartrate_intraday_features.reset_index(inplace=True)
return heartrate_intraday_features return heartrate_intraday_features