update
This commit is contained in:
parent
5d7436ffe9
commit
e92ad61864
@ -1,103 +0,0 @@
|
|||||||
{
|
|
||||||
"cells": [
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": 3,
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"ename": "NameError",
|
|
||||||
"evalue": "name 'text_data' is not defined",
|
|
||||||
"output_type": "error",
|
|
||||||
"traceback": [
|
|
||||||
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
|
|
||||||
"\u001b[0;31mNameError\u001b[0m Traceback (most recent call last)",
|
|
||||||
"\u001b[0;32m<ipython-input-3-d179e01d96de>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m 13\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 14\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 15\u001b[0;31m \u001b[0mw2v_model\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mgensim\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmodels\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mWord2Vec\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mtext_data\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0msize\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m300\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mmin_count\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m1\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mwindow\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m5\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0miter\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m50\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m 16\u001b[0m \u001b[0mw2v_model\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwv\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'word'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 17\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
|
|
||||||
"\u001b[0;31mNameError\u001b[0m: name 'text_data' is not defined"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"source": [
|
|
||||||
"from nltk.tokenize import sent_tokenize, word_tokenize\n",
|
|
||||||
"import warnings\n",
|
|
||||||
" \n",
|
|
||||||
"warnings.filterwarnings(action = 'ignore')\n",
|
|
||||||
" \n",
|
|
||||||
"import gensim\n",
|
|
||||||
"from gensim.models import Word2Vec\n",
|
|
||||||
"from sklearn.feature_extraction.text import TfidfVectorizer\n",
|
|
||||||
"import numpy as np\n",
|
|
||||||
"from sklearn.datasets import load_iris\n",
|
|
||||||
"from sklearn.model_selection import train_test_split\n",
|
|
||||||
"from sklearn.naive_bayes import GaussianNB\n",
|
|
||||||
"\n",
|
|
||||||
"sample = open(\"/train/in.tsv\", \"r\")\n",
|
|
||||||
"s = sample.read()\n",
|
|
||||||
" \n",
|
|
||||||
"# Replaces escape character with space\n",
|
|
||||||
"f = s.replace(\"\\n\", \" \")\n",
|
|
||||||
" \n",
|
|
||||||
"data = []\n",
|
|
||||||
" \n",
|
|
||||||
"# iterate through each sentence in the file\n",
|
|
||||||
"for i in sent_tokenize(f):\n",
|
|
||||||
" temp = []\n",
|
|
||||||
" \n",
|
|
||||||
" # tokenize the sentence into words\n",
|
|
||||||
" for j in word_tokenize(i):\n",
|
|
||||||
" temp.append(j.lower())\n",
|
|
||||||
" \n",
|
|
||||||
" data.append(temp)\n",
|
|
||||||
" \n",
|
|
||||||
"# Create CBOW model\n",
|
|
||||||
"model1 = gensim.models.Word2Vec(data, min_count = 1, \n",
|
|
||||||
" size = 100, window = 5)\n",
|
|
||||||
" \n",
|
|
||||||
"w2v_model = gensim.models.Word2Vec(text_data, size=300, min_count=1, window=5, iter=50)\n",
|
|
||||||
"w2v_model.wv['word']\n",
|
|
||||||
"\n",
|
|
||||||
"with open(\"train/in.tsv\") as f:\n",
|
|
||||||
" content = f.readlines()\n",
|
|
||||||
" with open(\"train/expected.tsv\") as ff:\n",
|
|
||||||
" y = ff.readlines()\n",
|
|
||||||
" vectorizer = TfidfVectorizer(ngram_range=(1,2), use_idf = False)\n",
|
|
||||||
" vectorizer = TfidfVectorizer()\n",
|
|
||||||
" x = vectorizer.fit_transform(content)\n",
|
|
||||||
" x=x.toarray()\n",
|
|
||||||
" y=y.toarray()\n",
|
|
||||||
" model = GaussianNB()\n",
|
|
||||||
" model.fit(x,y)\n",
|
|
||||||
" y_pred = model.predict([[0,1]])\n",
|
|
||||||
" print(y_pred)"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": null,
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [],
|
|
||||||
"source": []
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"metadata": {
|
|
||||||
"kernelspec": {
|
|
||||||
"display_name": "Python 3",
|
|
||||||
"language": "python",
|
|
||||||
"name": "python3"
|
|
||||||
},
|
|
||||||
"language_info": {
|
|
||||||
"codemirror_mode": {
|
|
||||||
"name": "ipython",
|
|
||||||
"version": 3
|
|
||||||
},
|
|
||||||
"file_extension": ".py",
|
|
||||||
"mimetype": "text/x-python",
|
|
||||||
"name": "python",
|
|
||||||
"nbconvert_exporter": "python",
|
|
||||||
"pygments_lexer": "ipython3",
|
|
||||||
"version": "3.8.3"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"nbformat": 4,
|
|
||||||
"nbformat_minor": 4
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user