|
|
@ -1528,7 +1528,7 @@
|
|
|
|
"source": [
|
|
|
|
"source": [
|
|
|
|
"cm_w2v = confusion_matrix(y_test_word2vec, y_predicted_word2vec)\n",
|
|
|
|
"cm_w2v = confusion_matrix(y_test_word2vec, y_predicted_word2vec)\n",
|
|
|
|
"fig = plt.figure(figsize=(10, 10))\n",
|
|
|
|
"fig = plt.figure(figsize=(10, 10))\n",
|
|
|
|
"plot = plot_confusion_matrix(cm, classes=['Irrelevant','Disaster','Unsure'], normalize=False, title='Confusion matrix')\n",
|
|
|
|
"plot = plot_confusion_matrix(cm_w2v, classes=['Irrelevant','Disaster','Unsure'], normalize=False, title='Confusion matrix')\n",
|
|
|
|
"plt.show()\n",
|
|
|
|
"plt.show()\n",
|
|
|
|
"print(\"Word2Vec confusion matrix\")\n",
|
|
|
|
"print(\"Word2Vec confusion matrix\")\n",
|
|
|
|
"print(cm_w2v)\n",
|
|
|
|
"print(cm_w2v)\n",
|
|
|
|