2.1.3.estimator.ipynb의 마지막 문장 결과오류 출력문제

[kslee5838]

2.1.3.estimator.ipynb

의 아래의 마지막 코딩에서 오류가 납니다.
estimator.train(train_input_fn)

~~마지막 에러내용은 이렇게 생겼네요. 해결책이 없을 까요?
AttributeError: 'ShuffleDataset' object has no attribute 'make_one_shot_iterator'