Геометрический train_test_split_edges PyTorch, реализованный в наборе данных с граничными функциями, приводит к сбою блокнота google-colab
Привет, сообщество stackoverflow,
Я новичок в PyTorch Geometric. Пока я разбивал свой набор данных (размер данных => Данные (x=[14254, 1647], edge_index=[2, 8552], edge_attr=[8552, 8])), используя train_test_split_edges из геометрического pytorch, блокнот google-colab падает без любое указание на причину этого крушения. Мои вопросы :
- Почему это происходит ?
- Как я могу это исправить ?
- Должен ли я реализовать собственный train_test_split_edges для моего большого набора данных?
- Как я могу реализовать собственный train_test_split_edges?
С уважением, Чихеб