search

ChangwenXu98 / TransPolymer

Implementation of "TransPolymer: a Transformer-based language model for polymer property predictions" in PyTorch
MIT License
53 stars 19 forks source link

Issues with the running of Downsteam.py #13

Closed cxliangUT closed 8 months ago

cxliangUT commented 10 months ago

Hello, I successfully prepared the environment, and the pre-training goes pretty well, however, when I begin to run the Downstream file, I encountered this issue (attched in the output.txt).

output.txt

There could be some issues with line 364 of the code: train_data.iloc[:, 1] = scaler.fit_transform(train_data.iloc[:, 1].values.reshape(-1, 1))

I output the shape of the train_data and test_data, it is (2,1)

ChangwenXu98 commented 9 months ago

Hi @cxliangUT, sorry for late reply. there are two columns in the input file, but the shape of train data and test data turns out to be 1-column. Have you modified the code about reading csv files? And is it possible that the files in your repo are incomplete?