{ "cells": [ { "cell_type": "code", "execution_count": 13, "id": "8f925fb7-86ba-487f-ab85-88754d777860", "metadata": { "tags": [] }, "outputs": [], "source": [ "import json\n", "with open(\"text/text_label.json\",\"r\",encoding='utf-8') as f:\n", " text_label = json.load(f)" ] }, { "cell_type": "code", "execution_count": 14, "id": "d2c0a048-1db7-4236-9f26-539ed31d3d27", "metadata": { "tags": [] }, "outputs": [], "source": [ "import random\n", "random.seed(0)\n", "def post_process(model_output,text_label):\n", " text_list = text_label[model_output]\n", " text,sent = random.sample(text_list,1)[0]\n", " return {'label' : model_output,\n", " 'text' : text,\n", " 'sentiment' : sent}" ] }, { "cell_type": "code", "execution_count": 15, "id": "f8ca0ad8-bc0c-4766-8e13-fe093c5290df", "metadata": { "tags": [] }, "outputs": [ { "data": { "text/plain": [ "{'label': 'bark', 'text': '아니야, 아니야!', 'sentiment': '부정'}" ] }, "execution_count": 15, "metadata": {}, "output_type": "execute_result" } ], "source": [ "model_output = 'bark'\n", "post_process(model_output,text_label)" ] }, { "cell_type": "code", "execution_count": null, "id": "da690a64-4dea-4b2a-89c1-23ea8bad955c", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "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.16" } }, "nbformat": 4, "nbformat_minor": 5 }