From 3132c5a752430cde89068c0f6ca0aa738ada075a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Assun=C3=A7ao=20Jeshon?= <jeshon.assuncao@he-arc.ch>
Date: Fri, 16 Nov 2018 13:31:12 +0100
Subject: [PATCH] Change derivate to gradient

---
 load_data.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/load_data.py b/load_data.py
index 45efa85..5f86b7c 100644
--- a/load_data.py
+++ b/load_data.py
@@ -367,10 +367,15 @@ def pretreat(
         df_filter = df[~pd.isna(df['impedance_nb'])]
     df = df_filter
 
-    # Derivate X values
+    # # Derivate X values
+    # if derivate:
+    #     df[X_labels] = df[X_labels].diff()
+    #     df = df.iloc[1:]
+
+    # Gradient X values
     if derivate:
-        df[X_labels] = df[X_labels].diff()
-        df = df.iloc[1:]
+        for x_label in X_labels:
+            df[x_label] = np.gradient(df[x_label].values)
 
     # Normalize X features
     if fit_scaler:
-- 
GitLab