bugfix

div by 0
reeteshsudhakar · Dec 5, 2023 · bd24d9a · bd24d9a
1 parent 76ef0a0
commit bd24d9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/data-processing.ipynb b/src/data-processing.ipynb
@@ -69,7 +69,7 @@
     "                   \"PRODUCT_COMBINATION\", \"DAYS_FIRST_DRAWING\", \"DAYS_FIRST_DUE\", \"DAYS_LAST_DUE_1ST_VERSION\", \"DAYS_LAST_DUE\", \"DAYS_TERMINATION\"], inplace=True)\n",
     "prev = prev[prev[\"NAME_CONTRACT_STATUS\"] != \"Canceled\"]\n",
     "prev.fillna(0, inplace=True)\n",
-    "prev[\"TIME_TO_PAY\"] = prev[\"AMT_CREDIT\"] / prev[\"AMT_ANNUITY\"]\n",
+    "prev[\"TIME_TO_PAY\"] = np.where(prev[\"AMT_ANNUITY\"] == 0, 0, prev[\"AMT_CREDIT\"] / prev[\"AMT_ANNUITY\"])\n",
     "prev[\"AMT_CREDIT\"] = np.where(prev[\"NAME_CONTRACT_STATUS\"] == \"Refused\", 0, prev[\"AMT_CREDIT\"])\n",
     "prev[\"CREDIT_DIFFERENCE\"] = prev[\"AMT_CREDIT\"] - prev[\"AMT_APPLICATION\"]\n",
     "prev.drop(columns=[\"AMT_ANNUITY\", \"AMT_APPLICATION\", \"AMT_CREDIT\"], inplace=True)\n",