From 48591a4bd399f79d5ce29e90d849ca7db415b07b Mon Sep 17 00:00:00 2001 From: Meng Li <34143965+Meng6@users.noreply.github.com> Date: Fri, 12 Jun 2020 13:28:41 -0400 Subject: [PATCH] Add exception for screen feature extraction --- src/features/screen/screen_base.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/features/screen/screen_base.py b/src/features/screen/screen_base.py index d274fc29..d4618f85 100644 --- a/src/features/screen/screen_base.py +++ b/src/features/screen/screen_base.py @@ -9,7 +9,12 @@ def getEpisodeDurationFeatures(screen_data, day_segment, episode, features, phon duration_helper = pd.concat([duration_helper, screen_data_episode[["time_diff"]].groupby(["local_start_date"]).count().rename(columns = {"time_diff": "screen_" + day_segment + "_countepisode" + episode})], axis = 1) if "episodepersensedminutes" in features: for date, row in screen_data_episode[["time_diff"]].groupby(["local_start_date"]).count().iterrows(): - sensed_minutes = phone_sensed_bins.loc[date, :].sum() * bin_size + + try: + sensed_minutes = phone_sensed_bins.loc[date, :].sum() * bin_size + except: + raise ValueError("You need to include the screen sensor in the list for phone_sensed_bins.") + episode_per_sensedminutes = row["time_diff"] / (1 if sensed_minutes == 0 else sensed_minutes) duration_helper.loc[date, "screen_" + day_segment + "_episodepersensedminutes" + episode] = episode_per_sensedminutes if "sumduration" in features: