junos
/
stress_at_work_analysis
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a getter for communication data.

rapids
junos 2021-08-19 17:36:26 +02:00
parent 93f136b080
commit 97c693d252
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
machine_learning/pipeline.py
View File

@ -50,6 +50,8 @@ class SensorFeatures:
def get_sensor_data(self, data_type) -> pd.DataFrame: def get_sensor_data(self, data_type) -> pd.DataFrame:
if data_type == "proximity": if data_type == "proximity":
return self.df_proximity return self.df_proximity
elif data_type == "communication":
return self.df_calls_sms
else: else:
raise KeyError("This data type has not been implemented.") raise KeyError("This data type has not been implemented.")