[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-15 10:09:50 +03:00
parent 11e5dc3b87
commit 12679a907e

@ -303,24 +303,39 @@ def kernel_3(
def kernel_4( def kernel_4(
o_2, o_2,
input_texts=None,
): ):
import keras.preprocessing.sequence import keras.preprocessing.sequence
t1 = 'blahb blahb blah' if input_texts is None:
t2 = o_2['token'].texts_to_sequences( input_texts = [
[t1], 'blahb blahb blah'
) ]
t3 = keras.preprocessing.sequence.pad_sequences(
t2, t6 = []
maxlen=o_2['max_len'] for o in input_texts:
) t1 = o
t4 = o_2['model'].predict( t2 = o_2['token'].texts_to_sequences(
t3, [t1],
) )
t3 = keras.preprocessing.sequence.pad_sequences(
t2,
maxlen=o_2['max_len']
)
t4 = o_2['model'].predict(
t3,
)
t6.append(
dict(
text=o,
score=t4[0],
)
)
return dict( return dict(
t1=t1, t1=t1,
t2=t2, t2=t2,
t3=t3, t3=t3,
t4=t4, t4=t4,
t6=t6,
) )