Для Процент кликов (CTR) , я хочу запустить FieldAwareNeuralFactorizationMachineModel (fnfm) в Набор данных Avazu .
После кода в examples/main.py (https://github.com/rixwew/pytorch-fm/blob/master/examples/main.py), и я успешно строю модель fnfm на основе набора поездов.
Однако я не знаю, как использовать эту модель для прогнозирования на тестовом наборе (для передачи в Kaggle). В частности, я использовал следующее команда для чтения набора тестов
testset = AvazuDataset(path_to_testset, rebuild_cache=True)
Экран выводит:
"Create avazu dataset cache: counting features: : 4577464it [00:07, 644823.68it/s]
Create avazu dataset cache: setup lmdb: : 4577464it [00:07, 639765.05it/s] "
Я думал, что набор тестов был успешно загружен, но когда я проверил len(testset), он равен 0. Кажется, что тестовый набор не удалось успешно загрузить, поэтому я не смог сделать прогноз для него.
Мой вопрос: как использовать эту библиотеку для прогнозирования на тестовом наборе Avazu?
Я боролся с этим в течение приблизительно 2 дней, но не мог понять, как решить это.
Большое спасибо