From 46d6f281147c23700fa0e92095b7a9fe61965100 Mon Sep 17 00:00:00 2001 From: s459312 Date: Thu, 25 May 2023 14:18:57 +0200 Subject: [PATCH] changed database + some fixes --- DialogManager.ipynb | 75 +++++++++++++++++++++++++-------------------- 1 file changed, 42 insertions(+), 33 deletions(-) diff --git a/DialogManager.ipynb b/DialogManager.ipynb index 864972d..5a5e99a 100644 --- a/DialogManager.ipynb +++ b/DialogManager.ipynb @@ -2,7 +2,7 @@ "cells": [ { "cell_type": "code", - "execution_count": 57, + "execution_count": 75, "metadata": {}, "outputs": [], "source": [ @@ -38,58 +38,58 @@ " if slot not in self.state['request_state'][domain]:\n", " self.state['request_state'][domain][slot] = 0\n", " \n", - " self.state['user_action'] = user_act\n", + " self.state['user_act'] = user_act\n", " return self.state" ] }, { "cell_type": "code", - "execution_count": 58, + "execution_count": 109, "metadata": {}, "outputs": [], "source": [ "dst = DST()\n", - "user_act = [('inform', 'payment', 'type', 'credit card'), ('inform', 'product', 'name', 'iPhone'), ('request', 'product', 'type', 'warzywo')]\n", + "user_act = [('inform', 'payment', 'type', 'karta'), ('inform', 'delivery', 'type','paczkomat'), ('inform', 'product', 'type', 'telefon'), ('request', 'product', 'name', '?')]\n", "state = dst.update(user_act)" ] }, { "cell_type": "code", - "execution_count": 59, + "execution_count": 110, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ - "{'payment': {'type': 'credit card', 'amount': '', 'loyalty_card': ''}, 'delivery': {'type': '', 'address': '', 'time': ''}, 'product': {'name': 'iPhone', 'type': '', 'brand': '', 'price': '', 'quantity': '', 'quality': ''}}\n", - "{'product': {'type': 0}}\n" + "{'payment': {'type': 'karta', 'amount': '', 'loyalty_card': ''}, 'delivery': {'type': 'paczkomat', 'address': '', 'time': ''}, 'product': {'name': '', 'type': 'telefon', 'brand': '', 'price': '', 'quantity': '', 'quality': ''}}\n", + "{'product': {'name': 0}}\n" ] }, { "data": { "text/plain": [ - "{'user_act': [],\n", + "{'user_act': [('inform', 'payment', 'type', 'karta'),\n", + " ('inform', 'delivery', 'type', 'paczkomat'),\n", + " ('inform', 'product', 'type', 'telefon'),\n", + " ('request', 'product', 'name', '?')],\n", " 'system_act': [],\n", - " 'belief_state': {'payment': {'type': 'credit card',\n", + " 'belief_state': {'payment': {'type': 'karta',\n", " 'amount': '',\n", " 'loyalty_card': ''},\n", - " 'delivery': {'type': '', 'address': '', 'time': ''},\n", - " 'product': {'name': 'iPhone',\n", - " 'type': '',\n", + " 'delivery': {'type': 'paczkomat', 'address': '', 'time': ''},\n", + " 'product': {'name': '',\n", + " 'type': 'telefon',\n", " 'brand': '',\n", " 'price': '',\n", " 'quantity': '',\n", " 'quality': ''}},\n", - " 'request_state': {'product': {'type': 0}},\n", + " 'request_state': {'product': {'name': 0}},\n", " 'terminated': False,\n", - " 'history': [],\n", - " 'user_action': [('inform', 'payment', 'type', 'credit card'),\n", - " ('inform', 'product', 'name', 'iPhone'),\n", - " ('request', 'product', 'type', 'warzywo')]}" + " 'history': []}" ] }, - "execution_count": 59, + "execution_count": 110, "metadata": {}, "output_type": "execute_result" } @@ -102,39 +102,37 @@ }, { "cell_type": "code", - "execution_count": 60, + "execution_count": 122, "metadata": {}, "outputs": [], "source": [ "from collections import defaultdict\n", - "import copy\n", "import json\n", - "from copy import deepcopy\n", + "\n", "\n", "\n", "\n", "class SimpleRulePolicy():\n", " def __init__(self):\n", - " self.db = json.load(open('product_db.json'))\n", + " with open('product_db.json', encoding='utf-8') as json_file:\n", + " self.db = json.load(json_file)\n", "\n", " def predict(self, state):\n", " self.results = []\n", " system_action = defaultdict(list)\n", " user_action = defaultdict(list)\n", - "\n", - " for intent, domain, slot, value in state['user_action']:\n", + " system_acts = []\n", + " for intent, domain, slot, value in state['user_act']:\n", " user_action[(domain.lower(), intent.lower())].append((slot.lower(), value))\n", - "\n", + " print(user_action)\n", " for user_act in user_action:\n", " self.update_system_action(user_act, user_action, state, system_action)\n", - "\n", - " # ReguĊ‚a 3\n", - " \n", - "\n", + " print(self.update_system_action(user_act, user_action, state, system_action))\n", " system_acts = [[intent, domain, slot, value] for (domain, intent), slots in system_action.items() for slot, value in slots]\n", - " state['system_action'] = system_acts\n", + " state['system_act'] = system_acts\n", " return system_acts\n", "\n", + "\n", " def update_system_action(self, user_act, user_action, state, system_action):\n", " domain, intent = user_act\n", "\n", @@ -155,22 +153,33 @@ " system_action[(domain, 'Inform')].append(['Choice', str(len(self.results))])\n", " choice = self.results[0]\n", "\n", - " if domain in [\"product\"]:\n", + " if domain in [\"payment\", \"delivery\", \"product\"]:\n", " system_action[(domain, 'Recommend')].append(['Name', choice['name']])" ] }, { "cell_type": "code", - "execution_count": 61, + "execution_count": 123, "metadata": {}, "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "defaultdict(, {('payment', 'inform'): [('type', 'karta')], ('delivery', 'inform'): [('type', 'paczkomat')], ('product', 'inform'): [('type', 'telefon')], ('product', 'request'): [('name', '?')]})\n", + "None\n", + "None\n", + "None\n", + "None\n" + ] + }, { "data": { "text/plain": [ "[]" ] }, - "execution_count": 61, + "execution_count": 123, "metadata": {}, "output_type": "execute_result" }