You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

fastnlp_10tmin_tutorial.rst 12 kB

7 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375
  1. fastNLP上手教程
  2. ===============
  3. fastNLP提供方便的数据预处理,训练和测试模型的功能
  4. DataSet & Instance
  5. ------------------
  6. fastNLP用DataSet和Instance保存和处理数据。每个DataSet表示一个数据集,每个Instance表示一个数据样本。一个DataSet存有多个Instance,每个Instance可以自定义存哪些内容。
  7. 有一些read\_\*方法,可以轻松从文件读取数据,存成DataSet。
  8. .. code:: ipython3
  9. from fastNLP import DataSet
  10. from fastNLP import Instance
  11. # 从csv读取数据到DataSet
  12. win_path = "C:\\Users\zyfeng\Desktop\FudanNLP\\fastNLP\\test\\data_for_tests\\tutorial_sample_dataset.csv"
  13. dataset = DataSet.read_csv(win_path, headers=('raw_sentence', 'label'), sep='\t')
  14. print(dataset[0])
  15. .. parsed-literal::
  16. {'raw_sentence': A series of escapades demonstrating the adage that what is good for the goose is also good for the gander , some of which occasionally amuses but none of which amounts to much of a story .,
  17. 'label': 1}
  18. .. code:: ipython3
  19. # DataSet.append(Instance)加入新数据
  20. dataset.append(Instance(raw_sentence='fake data', label='0'))
  21. dataset[-1]
  22. .. parsed-literal::
  23. {'raw_sentence': fake data,
  24. 'label': 0}
  25. .. code:: ipython3
  26. # DataSet.apply(func, new_field_name)对数据预处理
  27. # 将所有数字转为小写
  28. dataset.apply(lambda x: x['raw_sentence'].lower(), new_field_name='raw_sentence')
  29. # label转int
  30. dataset.apply(lambda x: int(x['label']), new_field_name='label_seq', is_target=True)
  31. # 使用空格分割句子
  32. dataset.drop(lambda x: len(x['raw_sentence'].split()) == 0)
  33. def split_sent(ins):
  34. return ins['raw_sentence'].split()
  35. dataset.apply(split_sent, new_field_name='words', is_input=True)
  36. .. code:: ipython3
  37. # DataSet.drop(func)筛除数据
  38. # 删除低于某个长度的词语
  39. dataset.drop(lambda x: len(x['words']) <= 3)
  40. .. code:: ipython3
  41. # 分出测试集、训练集
  42. test_data, train_data = dataset.split(0.3)
  43. print("Train size: ", len(test_data))
  44. print("Test size: ", len(train_data))
  45. .. parsed-literal::
  46. Train size: 54
  47. Test size:
  48. Vocabulary
  49. ----------
  50. fastNLP中的Vocabulary轻松构建词表,将词转成数字
  51. .. code:: ipython3
  52. from fastNLP import Vocabulary
  53. # 构建词表, Vocabulary.add(word)
  54. vocab = Vocabulary(min_freq=2)
  55. train_data.apply(lambda x: [vocab.add(word) for word in x['words']])
  56. vocab.build_vocab()
  57. # index句子, Vocabulary.to_index(word)
  58. train_data.apply(lambda x: [vocab.to_index(word) for word in x['words']], new_field_name='word_seq', is_input=True)
  59. test_data.apply(lambda x: [vocab.to_index(word) for word in x['words']], new_field_name='word_seq', is_input=True)
  60. print(test_data[0])
  61. .. parsed-literal::
  62. {'raw_sentence': the plot is romantic comedy boilerplate from start to finish .,
  63. 'label': 2,
  64. 'label_seq': 2,
  65. 'words': ['the', 'plot', 'is', 'romantic', 'comedy', 'boilerplate', 'from', 'start', 'to', 'finish', '.'],
  66. 'word_seq': [2, 13, 9, 24, 25, 26, 15, 27, 11, 28, 3]}
  67. .. code:: ipython3
  68. # 假设你们需要做强化学习或者gan之类的项目,也许你们可以使用这里的dataset
  69. from fastNLP.core.batch import Batch
  70. from fastNLP.core.sampler import RandomSampler
  71. batch_iterator = Batch(dataset=train_data, batch_size=2, sampler=RandomSampler())
  72. for batch_x, batch_y in batch_iterator:
  73. print("batch_x has: ", batch_x)
  74. print("batch_y has: ", batch_y)
  75. break
  76. .. parsed-literal::
  77. batch_x has: {'words': array([list(['this', 'kind', 'of', 'hands-on', 'storytelling', 'is', 'ultimately', 'what', 'makes', 'shanghai', 'ghetto', 'move', 'beyond', 'a', 'good', ',', 'dry', ',', 'reliable', 'textbook', 'and', 'what', 'allows', 'it', 'to', 'rank', 'with', 'its', 'worthy', 'predecessors', '.']),
  78. list(['the', 'entire', 'movie', 'is', 'filled', 'with', 'deja', 'vu', 'moments', '.'])],
  79. dtype=object), 'word_seq': tensor([[ 19, 184, 6, 1, 481, 9, 206, 50, 91, 1210, 1609, 1330,
  80. 495, 5, 63, 4, 1269, 4, 1, 1184, 7, 50, 1050, 10,
  81. 8, 1611, 16, 21, 1039, 1, 2],
  82. [ 3, 711, 22, 9, 1282, 16, 2482, 2483, 200, 2, 0, 0,
  83. 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
  84. 0, 0, 0, 0, 0, 0, 0]])}
  85. batch_y has: {'label_seq': tensor([3, 2])}
  86. Model
  87. -----
  88. .. code:: ipython3
  89. # 定义一个简单的Pytorch模型
  90. from fastNLP.models import CNNText
  91. model = CNNText(embed_num=len(vocab), embed_dim=50, num_classes=5, padding=2, dropout=0.1)
  92. model
  93. .. parsed-literal::
  94. CNNText(
  95. (embed): Embedding(
  96. (embed): Embedding(77, 50, padding_idx=0)
  97. (dropout): Dropout(p=0.0)
  98. )
  99. (conv_pool): ConvMaxpool(
  100. (convs): ModuleList(
  101. (0): Conv1d(50, 3, kernel_size=(3,), stride=(1,), padding=(2,))
  102. (1): Conv1d(50, 4, kernel_size=(4,), stride=(1,), padding=(2,))
  103. (2): Conv1d(50, 5, kernel_size=(5,), stride=(1,), padding=(2,))
  104. )
  105. )
  106. (dropout): Dropout(p=0.1)
  107. (fc): Linear(
  108. (linear): Linear(in_features=12, out_features=5, bias=True)
  109. )
  110. )
  111. Trainer & Tester
  112. ----------------
  113. 使用fastNLP的Trainer训练模型
  114. .. code:: ipython3
  115. from fastNLP import Trainer
  116. from copy import deepcopy
  117. from fastNLP import CrossEntropyLoss
  118. from fastNLP import AccuracyMetric
  119. .. code:: ipython3
  120. # 进行overfitting测试
  121. copy_model = deepcopy(model)
  122. overfit_trainer = Trainer(model=copy_model,
  123. train_data=test_data,
  124. dev_data=test_data,
  125. loss=CrossEntropyLoss(pred="output", target="label_seq"),
  126. metrics=AccuracyMetric(),
  127. n_epochs=10,
  128. save_path=None)
  129. overfit_trainer.train()
  130. .. parsed-literal::
  131. training epochs started 2018-12-07 14:07:20
  132. .. parsed-literal::
  133. HBox(children=(IntProgress(value=0, layout=Layout(flex='2'), max=20), HTML(value='')), layout=Layout(display='…
  134. .. parsed-literal::
  135. Epoch 1/10. Step:2/20. AccuracyMetric: acc=0.037037
  136. Epoch 2/10. Step:4/20. AccuracyMetric: acc=0.296296
  137. Epoch 3/10. Step:6/20. AccuracyMetric: acc=0.333333
  138. Epoch 4/10. Step:8/20. AccuracyMetric: acc=0.555556
  139. Epoch 5/10. Step:10/20. AccuracyMetric: acc=0.611111
  140. Epoch 6/10. Step:12/20. AccuracyMetric: acc=0.481481
  141. Epoch 7/10. Step:14/20. AccuracyMetric: acc=0.62963
  142. Epoch 8/10. Step:16/20. AccuracyMetric: acc=0.685185
  143. Epoch 9/10. Step:18/20. AccuracyMetric: acc=0.722222
  144. Epoch 10/10. Step:20/20. AccuracyMetric: acc=0.777778
  145. .. code:: ipython3
  146. # 实例化Trainer,传入模型和数据,进行训练
  147. trainer = Trainer(model=model,
  148. train_data=train_data,
  149. dev_data=test_data,
  150. loss=CrossEntropyLoss(pred="output", target="label_seq"),
  151. metrics=AccuracyMetric(),
  152. n_epochs=5)
  153. trainer.train()
  154. print('Train finished!')
  155. .. parsed-literal::
  156. training epochs started 2018-12-07 14:08:10
  157. .. parsed-literal::
  158. HBox(children=(IntProgress(value=0, layout=Layout(flex='2'), max=5), HTML(value='')), layout=Layout(display='i…
  159. .. parsed-literal::
  160. Epoch 1/5. Step:1/5. AccuracyMetric: acc=0.037037
  161. Epoch 2/5. Step:2/5. AccuracyMetric: acc=0.037037
  162. Epoch 3/5. Step:3/5. AccuracyMetric: acc=0.037037
  163. Epoch 4/5. Step:4/5. AccuracyMetric: acc=0.185185
  164. Epoch 5/5. Step:5/5. AccuracyMetric: acc=0.240741
  165. Train finished!
  166. .. code:: ipython3
  167. from fastNLP import Tester
  168. tester = Tester(data=test_data, model=model, metrics=AccuracyMetric())
  169. acc = tester.test()
  170. .. parsed-literal::
  171. [tester]
  172. AccuracyMetric: acc=0.240741
  173. In summary
  174. ----------
  175. fastNLP Trainer的伪代码逻辑
  176. ---------------------------
  177. 1. 准备DataSet,假设DataSet中共有如下的fields
  178. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  179. ::
  180. ['raw_sentence', 'word_seq1', 'word_seq2', 'raw_label','label']
  181. 通过
  182. DataSet.set_input('word_seq1', word_seq2', flag=True)将'word_seq1', 'word_seq2'设置为input
  183. 通过
  184. DataSet.set_target('label', flag=True)将'label'设置为target
  185. 2. 初始化模型
  186. ~~~~~~~~~~~~~
  187. ::
  188. class Model(nn.Module):
  189. def __init__(self):
  190. xxx
  191. def forward(self, word_seq1, word_seq2):
  192. # (1) 这里使用的形参名必须和DataSet中的input field的名称对应。因为我们是通过形参名, 进行赋值的
  193. # (2) input field的数量可以多于这里的形参数量。但是不能少于。
  194. xxxx
  195. # 输出必须是一个dict
  196. 3. Trainer的训练过程
  197. ~~~~~~~~~~~~~~~~~~~~
  198. ::
  199. (1) 从DataSet中按照batch_size取出一个batch,调用Model.forward
  200. (2) 将 Model.forward的结果 与 标记为target的field 传入Losser当中。
  201. 由于每个人写的Model.forward的output的dict可能key并不一样,比如有人是{'pred':xxx}, {'output': xxx};
  202. 另外每个人将target可能也会设置为不同的名称, 比如有人是label, 有人设置为target;
  203. 为了解决以上的问题,我们的loss提供映射机制
  204. 比如CrossEntropyLosser的需要的输入是(prediction, target)。但是forward的output是{'output': xxx}; 'label'是target
  205. 那么初始化losser的时候写为CrossEntropyLosser(prediction='output', target='label')即可
  206. (3) 对于Metric是同理的
  207. Metric计算也是从 forward的结果中取值 与 设置target的field中取值。 也是可以通过映射找到对应的值
  208. 一些问题.
  209. ---------
  210. 1. DataSet中为什么需要设置input和target
  211. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  212. ::
  213. 只有被设置为input或者target的数据才会在train的过程中被取出来
  214. (1.1) 我们只会在设置为input的field中寻找传递给Model.forward的参数。
  215. (1.2) 我们在传递值给losser或者metric的时候会使用来自:
  216. (a)Model.forward的output
  217. (b)被设置为target的field
  218. 2. 我们是通过forwad中的形参名将DataSet中的field赋值给对应的参数
  219. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  220. ::
  221. (1.1) 构建模型过程中,
  222. 例如:
  223. DataSet中x,seq_lens是input,那么forward就应该是
  224. def forward(self, x, seq_lens):
  225. pass
  226. 我们是通过形参名称进行匹配的field的
  227. 1. 加载数据到DataSet
  228. ~~~~~~~~~~~~~~~~~~~~
  229. 2. 使用apply操作对DataSet进行预处理
  230. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  231. ::
  232. (2.1) 处理过程中将某些field设置为input,某些field设置为target
  233. 3. 构建模型
  234. ~~~~~~~~~~~
  235. ::
  236. (3.1) 构建模型过程中,需要注意forward函数的形参名需要和DataSet中设置为input的field名称是一致的。
  237. 例如:
  238. DataSet中x,seq_lens是input,那么forward就应该是
  239. def forward(self, x, seq_lens):
  240. pass
  241. 我们是通过形参名称进行匹配的field的
  242. (3.2) 模型的forward的output需要是dict类型的。
  243. 建议将输出设置为{"pred": xx}.