Skip to content

Commit 1c90da1

Browse files
authored
Update titanicprocessing
1 parent ae06169 commit 1c90da1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

titanicprocessing

+4
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,10 @@ while degerson < 0.85:
235235
deger['test_score'].mean()
236236
# To see what score I got
237237

238+
cfm = confusion_matrix(y_test, model.predict(X_test))
239+
sns.heatmap(cfm, annot = True, fmt = 'd')
240+
# To visualize confusion matrix
241+
238242
# --------------- Preprocessing and Fit of the Test Sample Part ------------------
239243

240244
test = pd.read_csv('/kaggle/input/titanic/test.csv')

0 commit comments

Comments
 (0)