ly119399
3 years ago
2 changed files with 105 additions and 5 deletions
Split View
Diff Options
-
+1 -1modelscope/models/nlp/space/dialog_state_tracking_model.py
-
+104 -4tests/pipelines/nlp/test_dialog_state_tracking.py
+ 1
- 1
modelscope/models/nlp/space/dialog_state_tracking_model.py
View File
| @@ -88,7 +88,7 @@ class DialogStateTrackingModel(Model): | |||
| if u != 0: | |||
| diag_state[slot][i] = u | |||
| print(outputs) | |||
| # print(outputs) | |||
| return { | |||
| 'inputs': inputs, | |||
+ 104
- 4
tests/pipelines/nlp/test_dialog_state_tracking.py
View File
| @@ -18,9 +18,102 @@ class DialogStateTrackingTest(unittest.TestCase): | |||
| test_case = [{ | |||
| 'utter': { | |||
| 'User-1': | |||
| "I'm looking for a place to stay. It needs to be a guesthouse and include free wifi." | |||
| 'am looking for a place to to stay that has cheap price range it should be in a type of hotel' | |||
| }, | |||
| 'history_states': [{}] | |||
| }, { | |||
| 'utter': { | |||
| 'User-1': | |||
| 'am looking for a place to to stay that has cheap price range it should be in a type of hotel', | |||
| 'System-1': | |||
| 'Okay, do you have a specific area you want to stay in?', | |||
| 'Dialog_Act-1': { | |||
| 'Hotel-Request': [['Area', '?']] | |||
| }, | |||
| 'User-2': | |||
| "no, i just need to make sure it's cheap. oh, and i need parking" | |||
| }, | |||
| 'history_states': [{}, { | |||
| 'taxi': { | |||
| 'book': { | |||
| 'booked': [] | |||
| }, | |||
| 'semi': { | |||
| 'leaveAt': '', | |||
| 'destination': '', | |||
| 'departure': '', | |||
| 'arriveBy': '' | |||
| } | |||
| }, | |||
| 'police': { | |||
| 'book': { | |||
| 'booked': [] | |||
| }, | |||
| 'semi': {} | |||
| }, | |||
| 'restaurant': { | |||
| 'book': { | |||
| 'booked': [], | |||
| 'people': '', | |||
| 'day': '', | |||
| 'time': '' | |||
| }, | |||
| 'semi': { | |||
| 'food': '', | |||
| 'pricerange': '', | |||
| 'name': '', | |||
| 'area': '' | |||
| } | |||
| }, | |||
| 'hospital': { | |||
| 'book': { | |||
| 'booked': [] | |||
| }, | |||
| 'semi': { | |||
| 'department': '' | |||
| } | |||
| }, | |||
| 'hotel': { | |||
| 'book': { | |||
| 'booked': [], | |||
| 'people': '', | |||
| 'day': '', | |||
| 'stay': '' | |||
| }, | |||
| 'semi': { | |||
| 'name': 'not mentioned', | |||
| 'area': 'not mentioned', | |||
| 'parking': 'not mentioned', | |||
| 'pricerange': 'cheap', | |||
| 'stars': 'not mentioned', | |||
| 'internet': 'not mentioned', | |||
| 'type': 'hotel' | |||
| } | |||
| }, | |||
| 'attraction': { | |||
| 'book': { | |||
| 'booked': [] | |||
| }, | |||
| 'semi': { | |||
| 'type': '', | |||
| 'name': '', | |||
| 'area': '' | |||
| } | |||
| }, | |||
| 'train': { | |||
| 'book': { | |||
| 'booked': [], | |||
| 'people': '' | |||
| }, | |||
| 'semi': { | |||
| 'leaveAt': '', | |||
| 'destination': '', | |||
| 'day': '', | |||
| 'arriveBy': '', | |||
| 'departure': '' | |||
| } | |||
| } | |||
| }, {}] | |||
| }] | |||
| def test_run(self): | |||
| @@ -29,12 +122,19 @@ class DialogStateTrackingTest(unittest.TestCase): | |||
| model = DialogStateTrackingModel(cache_path) | |||
| preprocessor = DialogStateTrackingPreprocessor(model_dir=cache_path) | |||
| pipeline1 = DialogStateTrackingPipeline( | |||
| model=model, preprocessor=preprocessor) | |||
| pipelines = [ | |||
| DialogStateTrackingPipeline( | |||
| model=model, preprocessor=preprocessor), | |||
| # pipeline( | |||
| # task=Tasks.dialog_state_tracking, | |||
| # model=model, | |||
| # preprocessor=preprocessor) | |||
| ] | |||
| history_states = {} | |||
| pipelines_len = len(pipelines) | |||
| for step, item in enumerate(self.test_case): | |||
| history_states = pipeline1(item) | |||
| history_states = pipelines[step % pipelines_len](item) | |||
| print(history_states) | |||
| @unittest.skip('test with snapshot_download') | |||