執行datasets.IMDB.splits()出錯
想請問一下 我在執行 train_data, test_data = datasets.IMDB.splits(TEXT, LABEL) 的時候發生以下錯誤 AttributeError: 'function' object has no attribute 'splits' 我發現IMDB印出來的型態是func 再查了一下官方的原始碼 發現IMDB有class和function兩種型態 想請問該如何修正這個問題? 
回答列表
-
2021/03/13 下午 09:17張維元 (WeiYuan)贊同數：1不贊同數：0留言數：0
嗨，你好
我猜你是新版本的 pytorch， IMDB 的用法有改變： ``` # import datasets from torchtext.datasets import IMDB train_iter = IMDB(split='train') def tokenize(label, line): return line.split() tokens = [] for label, line in train_iter: tokens += tokenize(label, line) ``` [文件](https://pytorch.org/text/stable/datasets.html) 在這邊。
嗨，你好，我是維元，持續在不同的平台發表對 #資料科學、 #網頁開發 或 #軟體職涯 相關的文章。如果對於內文有疑問都歡迎與我們進一步的交流，都可以追蹤 我的粉絲專頁 ヽ(●´∀`●)ﾉ