Python: how to compare two array in easiest way

Question

I am doing machine learning stuffs. I meet a problem these day, hope someone can help me.

I have two arrays [Array(Prediction) and Array(Labels)) with both same dimension [Shape: (128, 5)]. The first dimension is the index of each predict and label. The second dimension is the results associated with each predict and label

I want to output the accuracy Here is my code.

right_count = 0
for i in range(Prediction.shape[0])  # Foreach each predict/label
    if（np.array_equal(Prediction[i], Labels[i])):  # Compare each result
        right_count += 1

accuracy = float(right_count) / Prediction.shape[0]'

I just wonder whether there is a better way to simply the following code.

Thank you

1'' · Accepted Answer · 2017-02-09 18:52:19Z

4

That code can be simplified, for instance:

accuracy = (Prediction == Labels).all(axis=1).mean()

This will also run faster than the original version because more of the operations are done inside Numpy instead of in Python code.

answered Feb 9, 2017 at 18:52

1''

27.2k32 gold badges149 silver badges204 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python: how to compare two array in easiest way

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related