{"cells":[{"metadata":{"_uuid":"65147b5d-83f1-44f1-a5bf-b10eb7c840a2","_cell_guid":"441b4b56-ce8a-40ff-9167-b3cc9d3b20c4","trusted":true},"cell_type":"code","source":"!pip install -q pyicu\n!pip install -q pycld2\n!pip install -q polyglot\n!pip install -q textstat\n!pip install -q googletrans","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"31a7b308-4df8-4d2c-8920-3ce40527374f","_cell_guid":"6a6f3bec-1c1b-459d-801e-a8711678f743","trusted":true},"cell_type":"code","source":"import warnings\nwarnings.filterwarnings(\"ignore\")\n\nimport os\nimport gc\nimport re\nimport folium\nimport textstat\nfrom scipy import stats\nfrom colorama import Fore, Back, Style, init\n\nimport math\nimport numpy as np\nimport scipy as sp\nimport pandas as pd\n\nimport random\nimport networkx as nx\nfrom pandas import Timestamp\n\nfrom PIL import Image\nfrom IPython.display import SVG\nfrom keras.utils import model_to_dot\n\nimport requests\nfrom IPython.display import HTML\n\nimport seaborn as sns\nfrom tqdm import tqdm\nimport matplotlib.cm as cm\nimport matplotlib.pyplot as plt\n\ntqdm.pandas()\n\nimport plotly.express as px\nimport plotly.graph_objects as go\nimport plotly.figure_factory as ff\nfrom plotly.subplots import make_subplots\n\nimport transformers\nimport tensorflow as tf\n\nfrom tensorflow.keras.callbacks import Callback\nfrom sklearn.metrics import accuracy_score, roc_auc_score\nfrom tensorflow.keras.callbacks import ModelCheckpoint, ReduceLROnPlateau, CSVLogger\n\nfrom tensorflow.keras.models import Model\nfrom kaggle_datasets import KaggleDatasets\nfrom tensorflow.keras.optimizers import Adam\nfrom tokenizers import BertWordPieceTokenizer\nfrom tensorflow.keras.layers import Dense, Input, Dropout, Embedding\nfrom tensorflow.keras.layers import LSTM, GRU, Conv1D, SpatialDropout1D\n\nfrom tensorflow.keras import layers\nfrom tensorflow.keras import optimizers\nfrom tensorflow.keras import activations\nfrom tensorflow.keras import constraints\nfrom tensorflow.keras import initializers\nfrom tensorflow.keras import regularizers\n\nimport tensorflow.keras.backend as K\nfrom tensorflow.keras.layers import *\nfrom tensorflow.keras.optimizers import *\nfrom tensorflow.keras.activations import *\nfrom tensorflow.keras.constraints import *\nfrom tensorflow.keras.initializers import *\nfrom tensorflow.keras.regularizers import *\n\nfrom sklearn import metrics\nfrom sklearn.utils import shuffle\nfrom gensim.models import Word2Vec\nfrom sklearn.cluster import KMeans\nfrom sklearn.decomposition import PCA\nfrom sklearn.feature_extraction.text import TfidfVectorizer,\\\n                                            CountVectorizer,\\\n                                            HashingVectorizer\n\nfrom nltk.stem.wordnet import WordNetLemmatizer \nfrom nltk.tokenize import word_tokenize\nfrom nltk.tokenize import TweetTokenizer  \n\nimport nltk\nfrom textblob import TextBlob\n\nfrom nltk.corpus import wordnet\nfrom nltk.corpus import stopwords\nfrom googletrans import Translator\nfrom nltk import WordNetLemmatizer\nfrom polyglot.detect import Detector\nfrom nltk.stem import WordNetLemmatizer\nfrom wordcloud import WordCloud, STOPWORDS\nfrom nltk.sentiment.vader import SentimentIntensityAnalyzer\n\nstopword=set(STOPWORDS)\n\nlem = WordNetLemmatizer()\ntokenizer=TweetTokenizer()\n\nnp.random.seed(0)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"5b26b297-a4fc-46de-93ad-83a55c4e997c","_cell_guid":"ba57b2f3-faa0-4f30-ac58-cff471cef9ef","trusted":true},"cell_type":"code","source":"DATA_PATH = \"/kaggle/input/jigsaw-multilingual-toxic-comment-classification/\"\nos.listdir(DATA_PATH)\nTEST_PATH = DATA_PATH + \"test.csv\"\nVAL_PATH = DATA_PATH + \"validation.csv\"\nTRAIN_PATH = DATA_PATH + \"jigsaw-toxic-comment-train.csv\"\n\nval_data = pd.read_csv(VAL_PATH)\ntest_data = pd.read_csv(TEST_PATH)\ntrain_data = pd.read_csv(TRAIN_PATH)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"dda13559-de60-4a49-842a-205feac15cd6","_cell_guid":"4d5334b6-bff6-4037-b9d0-0eaa684c8eb2","trusted":true},"cell_type":"code","source":"val = val_data\ntrain = train_data\n\ndef clean(text):\n    text = text.fillna(\"fillna\").str.lower()\n    text = text.map(lambda x: re.sub('\\\\n',' ',str(x)))\n    text = text.map(lambda x: re.sub(\"\\[\\[User.*\",'',str(x)))\n    text = text.map(lambda x: re.sub(\"\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\",'',str(x)))\n    text = text.map(lambda x: re.sub(\"\\(http://.*?\\s\\(http://.*\\)\",'',str(x)))\n    return text\n\nval[\"comment_text\"] = clean(val[\"comment_text\"])\ntest_data[\"content\"] = clean(test_data[\"content\"])\ntrain[\"comment_text\"] = clean(train[\"comment_text\"])","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"9f81b065-da28-473d-b381-32244b19b000","_cell_guid":"c36d7b59-4aa6-45c2-a3c5-df494623b87f","trusted":true},"cell_type":"code","source":"class RocAucEvaluation(Callback):\n    def __init__(self, validation_data=(), interval=1):\n        super(Callback, self).__init__()\n\n        self.interval = interval\n        self.X_val, self.y_val = validation_data\n\n    def on_epoch_end(self, epoch, logs={}):\n        if epoch % self.interval == 0:\n            y_pred = self.model.predict(self.X_val, verbose=0)\n            score = roc_auc_score(self.y_val, y_pred)\n            print(\"\\n ROC-AUC - epoch: {:d} - score: {:.6f}\".format(epoch+1, score))","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"b8ab92fc-ae1e-4a79-99ab-a9a510db28c9","_cell_guid":"a1ec0662-1bad-444c-8f9f-2fb40e48f3fe","trusted":true},"cell_type":"code","source":"def fast_encode(texts, tokenizer, chunk_size=240, maxlen=512):\n    tokenizer.enable_truncation(max_length=maxlen)\n    tokenizer.enable_padding(max_length=maxlen)\n    all_ids = []\n    \n    for i in range(0, len(texts), chunk_size):\n        text_chunk = texts[i:i+chunk_size].tolist()\n        encs = tokenizer.encode_batch(text_chunk)\n        all_ids.extend([enc.ids for enc in encs])\n    \n    return np.array(all_ids)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"9c1548ed-e194-4d1a-ac6d-e6db353f18a4","_cell_guid":"bc20ef52-cba4-49d7-bf59-3383c9bede8f","trusted":true},"cell_type":"code","source":"AUTO = tf.data.experimental.AUTOTUNE\n\ntpu = tf.distribute.cluster_resolver.TPUClusterResolver()\ntf.config.experimental_connect_to_cluster(tpu)\ntf.tpu.experimental.initialize_tpu_system(tpu)\nstrategy = tf.distribute.experimental.TPUStrategy(tpu)\n\nGCS_DS_PATH = KaggleDatasets().get_gcs_path('jigsaw-multilingual-toxic-comment-classification')\n\nEPOCHS = 2\nBATCH_SIZE = 32 * strategy.num_replicas_in_sync","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"9abadbb2-5568-40e1-8a03-a7a5d78b1658","_cell_guid":"17e13eeb-3636-47f2-b57c-8ad97fa28070","trusted":true},"cell_type":"code","source":"tokenizer = transformers.DistilBertTokenizer.from_pretrained('distilbert-base-multilingual-cased')\n\nsave_path = '/kaggle/working/distilbert_base_uncased/'\nif not os.path.exists(save_path):\n    os.makedirs(save_path)\ntokenizer.save_pretrained(save_path)\n\nfast_tokenizer = BertWordPieceTokenizer('distilbert_base_uncased/vocab.txt', \n                                        lowercase=True)\nfast_tokenizer","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"600dc4dc-749f-405b-9f20-832746333aa4","_cell_guid":"78d55c60-d68b-44d7-894b-3885f1169bc7","trusted":true},"cell_type":"code","source":"x_train = fast_encode(train.comment_text.astype(str), \n                      fast_tokenizer, maxlen=512)\nx_valid = fast_encode(val_data.comment_text.astype(str).values, \n                      fast_tokenizer, maxlen=512)\nx_test = fast_encode(test_data.content.astype(str).values, \n                     fast_tokenizer, maxlen=512)\n\ny_valid = val.toxic.values\ny_train = train.toxic.values","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"181fb612-f690-4261-bab9-df3795720ef6","_cell_guid":"dabb1ff7-da32-4eef-8133-d027876fb142","trusted":true},"cell_type":"code","source":"print(x_train)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"aafdff16-cda2-4b64-90bd-afe65bfd2c3d","_cell_guid":"358ff38c-9e0a-4c0d-8382-322541238fe3","trusted":true},"cell_type":"code","source":"train.comment_text.astype(str)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"46637d76-1e37-47d0-bafa-4c88d7b0b8ff","_cell_guid":"8f63cf0c-6746-46c5-bf90-2f92cb0a1f34","trusted":true},"cell_type":"code","source":"print(x_train.shape)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"3578686f-232a-4051-838a-a3f70d0f00f2","_cell_guid":"e6259523-0fe4-4517-bf81-bc0582709629","trusted":true},"cell_type":"code","source":"train_dataset = (\n    tf.data.Dataset\n    .from_tensor_slices((x_train, y_train))\n    .repeat()\n    .shuffle(2048)\n    .batch(BATCH_SIZE)\n    .prefetch(AUTO)\n)\n\nvalid_dataset = (\n    tf.data.Dataset\n    .from_tensor_slices((x_valid, y_valid))\n    .batch(BATCH_SIZE)\n    .cache()\n    .prefetch(AUTO)\n)\n\ntest_dataset = (\n    tf.data.Dataset\n    .from_tensor_slices(x_test)\n    .batch(BATCH_SIZE)\n)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"d0ade8b7-5b61-431e-b4d4-2b7e6388ad98","_cell_guid":"7913ecb7-0a02-4bc4-b167-4aecb8e22b41","trusted":true},"cell_type":"code","source":"def build_distilbert_model(transformer, max_len=512):\n    input_word_ids = Input(shape=(max_len,), dtype=tf.int32, name=\"input_word_ids\")\n    sequence_output = transformer(input_word_ids)[0]\n    cls_token = sequence_output[:, 0, :]\n    cls_token = Dense(500, activation=\"elu\")(cls_token)\n    cls_token = Dropout(0.1)(cls_token)\n    out = Dense(1, activation='sigmoid')(cls_token)\n    \n    model = Model(inputs=input_word_ids, outputs=out)\n    \n    model.compile(Adam(lr=1.5e-5), \n                  loss='binary_crossentropy', \n                  metrics=['accuracy'])\n    \n    return model","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"dca15858-6101-4d11-82c3-2f55dafeb27a","_cell_guid":"bea1d067-86b9-420f-8fbf-069f9437eb2a","trusted":true},"cell_type":"code","source":"with strategy.scope():\n    transformer_layer = transformers.TFDistilBertModel.\\\n    from_pretrained('distilbert-base-multilingual-cased')\n    model_distilbert = build_distilbert_model(transformer_layer, max_len=512)\n\nmodel_distilbert.summary()","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"0b9a884a-0980-4b50-b41d-37ab8a9b7464","_cell_guid":"030b5ecf-bac4-4922-a2b2-b8110ac69285","trusted":true},"cell_type":"code","source":"def callback():\n    cb = []\n\n    reduceLROnPlat = ReduceLROnPlateau(monitor='val_loss',  \n                                    factor=0.3, patience=3, \n                                    verbose=1, mode='auto', \n                                    epsilon=0.0001, cooldown=1, min_lr=0.000001)\n    cb.append(reduceLROnPlat)\n    log = CSVLogger('log.csv')\n    cb.append(log)\n\n    RocAuc = RocAucEvaluation(validation_data=(x_valid, y_valid), interval=1)\n    cb.append(RocAuc)\n    \n    return cb","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"96f8e03a-e560-490f-ad8c-71a29c933a07","_cell_guid":"be05f3a2-b60d-49d1-9d12-97add57a017d","trusted":true},"cell_type":"code","source":"calls = callback()\nN_STEPS = x_train.shape[0] // BATCH_SIZE\ntrain_history = model_distilbert.fit(\n    train_dataset,\n    steps_per_epoch=N_STEPS,\n    validation_data=valid_dataset,\n    callbacks = calls,\n    epochs=EPOCHS\n)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"927ce8c8-b603-41c6-96b5-0fa450693f45","_cell_guid":"c530422c-b14f-4b0a-a5c8-94b1cbeac406","trusted":true},"cell_type":"code","source":"SVG(tf.keras.utils.model_to_dot(model_distilbert, dpi=70).create(prog='dot', format='svg'))","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"0ae2d16d-6057-4691-9b96-a45530f03b45","_cell_guid":"abff8bce-ee28-45a1-a38a-a841058661db","trusted":true},"cell_type":"code","source":"translator = Translator()\n\ndef visualize_model_preds(model, indices=[0, 17, 1, 24]):\n    comments = val_data.comment_text.loc[indices].values.tolist()\n    preds = model.predict(x_valid[indices].reshape(len(indices), -1))\n\n    for idx, i in enumerate(indices):\n        if y_valid[i] == 0:\n            label = \"Non-toxic\"\n            color = f'{Fore.GREEN}'\n            symbol = '\\u2714'\n        else:\n            label = \"Toxic\"\n            color = f'{Fore.RED}'\n            symbol = '\\u2716'\n\n        print('{}{} {}'.format(color, str(idx+1) + \". \" + label, symbol))\n        print(f'{Style.RESET_ALL}')\n        print(\"ORIGINAL\")\n        print(comments[idx]); print(\"\")\n        print(\"TRANSLATED\")\n        print(translator.translate(comments[idx]).text)\n        fig = go.Figure()\n        if list.index(sorted(preds[:, 0]), preds[idx][0]) > 1:\n            yl = [preds[idx][0], 1 - preds[idx][0]]\n        else:\n            yl = [1 - preds[idx][0], preds[idx][0]]\n        fig.add_trace(go.Bar(x=['Non-Toxic', 'Toxic'], y=yl, marker=dict(color=[\"seagreen\", \"indianred\"])))\n        fig.update_traces(name=comments[idx])\n        fig.update_layout(xaxis_title=\"Labels\", yaxis_title=\"Probability\", template=\"plotly_white\", title_text=\"Predictions for validation comment #{}\".format(idx+1))\n        fig.show()","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"c26ba69f-cd99-4e37-aeec-3e513ebc852b","_cell_guid":"16c19ac7-3581-4d07-b8bf-b33d3908471e","trusted":true},"cell_type":"code","source":"visualize_model_preds(model_distilbert)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"2b7ec1de-0eb8-4607-aff0-7b77fe814659","_cell_guid":"f45f9e19-c7d1-43ca-b939-a0c225e8dfd3","trusted":true},"cell_type":"code","source":"sub = pd.read_csv(DATA_PATH + 'sample_submission.csv')\nsub['toxic'] = model_distilbert.predict(test_dataset, verbose=1)\nsub.to_csv('submission.csv', index=False)","execution_count":0,"outputs":[]},{"metadata":{"_uuid":"02b0564c-8477-48cc-b87d-4e23168d74cf","_cell_guid":"24fb7bc3-fcf6-4c61-855e-f7f3b470048d","trusted":true},"cell_type":"code","source":"model_distilbert.save_weights('model.h5')","execution_count":0,"outputs":[]}],"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat":4,"nbformat_minor":4}