Untitled

mail@pastecode.io avatar
unknown
python
2 months ago
337 B
6
Indexable
Never
def find_image(row):
    to_predict1 = pd.concat([test_images_vector, 
                            pd.concat([row] * len(test_images_vector), 
                                      ignore_index=True)], axis=1)
    return test_images. iloc[np.argmax(Ridge_best.predict(to_predict1))][0]


test_text_vector.apply(find_image, axis=1)
Leave a Comment