Получение значений argmax из массива 3D - PullRequest
Новая
       16

Получение значений argmax из массива 3D

0 голосов
/ 21 ноября 2018

У меня есть 3 цикла, и я хотел бы получить лучшее значение степени и C из среднего значения.

step_size = np.logspace(-2,3,6) //C values
degree = [1,2,3,4,5] //degree values
d_no =0
end_result = np.zeros((5,6,5))
for d in degree:

    log_no = 0

    for item in step_size:

        fold_no= 0

        for train_index, vld_index in kf.split(X_train):
            svc_poly = SVC(kernel='poly', degree=d, C=item,probability=True)
            //some codes


            fold_no += 1
        log_no += 1
        //mean is calculated here
    d_no +=1
//end for loop

В конце end_result должна иметь матрицу 5x6x5.Состоит из среднего для каждой степени в каждом сгибе.Так как я могу применить argmax для этого 3D-массива?Я узнал, что если он находится в 2d массиве, я могу просто использовать, например, для: 

best_c = step_size[np.argmax(mean)]
print(best_c)

Так, как я могу сделать что-то подобное, чтобы получить лучшую степень и C с 3D-массивом?Заранее спасибо!

...