logo
Loading...

執行datasets.IMDB.splits()出錯 - Cupoy

想請問一下我在執行train_data, test_data = datasets.IMDB.spl...

執行datasets.IMDB.splits()出錯

2021/03/13 下午 04:54
艾柯
觀看數:26
回答數:1
收藏數:0

想請問一下 我在執行 train_data, test_data = datasets.IMDB.splits(TEXT, LABEL) 的時候發生以下錯誤 AttributeError: 'function' object has no attribute 'splits' 我發現IMDB印出來的型態是func 再查了一下官方的原始碼 發現IMDB有class和function兩種型態 想請問該如何修正這個問題? ![q.png](http://kwassistfile.cupoy.com/000001782AC4FE7D0000000C6375706F795F72656C656173655155455354/1615261010334/large)

回答列表

  • 2021/03/13 下午 09:17
    張維元 (WeiYuan)
    贊同數:1
    不贊同數:0
    留言數:1

    嗨,你好
    我猜你是新版本的 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) 在這邊。

    嗨,你好,我是維元,持續在不同的平台發表對 #資料科學、 #網頁開發 或 #軟體職涯 相關的文章。如果對於內文有疑問都歡迎與我們進一步的交流,都可以追蹤 我的粉絲專頁 ヽ(●´∀`●)ノ