'Pandas column of string array to Long tensor

I have a pandas df that looks like this:

Ln2
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]
[C, C, C, C, C, C, G, I, O, P, P, P, R, R, R, R, R, R]

43,000+ rows where each array's length varies from 5 to 20, and this is 1 column fyi, so each row is a single cell.

I want to one hot this column. It was suggested to use torch.nn.functional.one_hot, but it wants it to be a float, and an index tensor. Any suggestions would be greatly appreciated.

pandastensorone-hot-encodingpytorch-geometric


Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source

Related Questions