Spaces:
Sleeping
Sleeping
James McCool
commited on
Commit
·
229b9de
1
Parent(s):
e8c9dac
Update data type handling for salary, median, and ownership difference calculations in reassess_edge.py to improve memory efficiency and precision.
Browse files
global_func/reassess_edge.py
CHANGED
|
@@ -58,9 +58,9 @@ def reassess_edge(refactored_frame: pd.DataFrame, original_frame: pd.DataFrame,
|
|
| 58 |
refactored_df = refactored_frame.copy()
|
| 59 |
refactored_df = refactored_df.reset_index(drop=True)
|
| 60 |
|
| 61 |
-
refactored_df['salary_diff'] = refactored_df['salary'] - orig_df['salary']
|
| 62 |
-
refactored_df['median_diff'] = refactored_df['median'] - orig_df['median']
|
| 63 |
-
refactored_df['own_diff'] = refactored_df['Own'] - orig_df['Own']
|
| 64 |
|
| 65 |
change_mask = refactored_df[refactored_df['median_diff'] != 0]
|
| 66 |
|
|
|
|
| 58 |
refactored_df = refactored_frame.copy()
|
| 59 |
refactored_df = refactored_df.reset_index(drop=True)
|
| 60 |
|
| 61 |
+
refactored_df['salary_diff'] = (refactored_df['salary'] - orig_df['salary']).astype('int16')
|
| 62 |
+
refactored_df['median_diff'] = (refactored_df['median'] - orig_df['median']).astype('float32')
|
| 63 |
+
refactored_df['own_diff'] = (refactored_df['Own'] - orig_df['Own']).astype('float32')
|
| 64 |
|
| 65 |
change_mask = refactored_df[refactored_df['median_diff'] != 0]
|
| 66 |
|