Commit cf0ef1a8 authored by uoega's avatar uoega
Browse files

Merge branch 'master' of https://git.scc.kit.edu/uoega/zero-shot-action-recognition into master

parents f69ea599 e915996b
......@@ -79,7 +79,7 @@ sentences = ['drink water',
#read labels from txt file:
label_list = []
class_label_file = open('class_sentences_40_ver5.txt','r')
class_label_file = open('class_sentences_40_ver5_roberta_aug_4.txt','r')
for lines in class_label_file:
#label = class_label_file.readline()
#print("Zeile" , lines, ": ")
......@@ -102,7 +102,7 @@ sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask']
print(sentence_embeddings.shape)
sentence_embeddings = sentence_embeddings.numpy()
print(sentence_embeddings.shape)
sio.savemat('mean_language_embedding_new.mat', {'attributes':sentence_embeddings})
sio.savemat('class_sentences_40_ver5_roberta_aug_4.mat', {'attributes':sentence_embeddings})
#Perform pooling. In this case, mean pooling
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment