// File: /workspace/github_cpp_makefile_star2_O2/CastellanZhang[P]lambdaFM/./src/FTRL/ftrl_trainer.h
// Function: ftrl_trainer::trainPair(fm_sample&, fm_sample&, double)
void ftrl_trainer::trainPair(fm_sample& sample1, fm_sample& sample2, const double deltaNDCG)
{
    unordered_map<string, ftrl_model_unit*> theta;
    for(unordered_map<string, double>::iterator iter = sample1.x.begin(); iter != sample1.x.end(); ++iter)
    {
        const string& index = iter->first;
        if(theta.find(index) == theta.end())
        {
            theta[index] = pModel->getOrInitModelUnit(index);
        }
    }
    for(unordered_map<string, double>::iterator iter = sample2.x.begin(); iter != sample2.x.end(); ++iter)
    {
        const string& index = iter->first;
        if(theta.find(index) == theta.end())
        {
            theta[index] = pModel->getOrInitModelUnit(index);
        }
    }
    //update w via FTRL
    if(k1)
    {
        for(unordered_map<string, ftrl_model_unit*>::iterator iter = theta.begin(); iter != theta.end(); ++iter)
        {
            ftrl_model_unit& mu = *(iter->second);
            mu.mtx.lock();
            if(fabs(mu.w_zi) <= w_l1)
            {
                mu.wi = 0.0;
            }
            else
            {
                if(force_v_sparse && mu.w_ni > 0 && 0.0 == mu.wi)
                {
                    mu.reinit_vi(pModel->init_mean, pModel->init_stdev);
                }
                mu.wi = (-1) *
                    (1 / (w_l2 + (w_beta + sqrt(mu.w_ni)) / w_alpha)) *
                    (mu.w_zi - utils::sgn(mu.w_zi) * w_l1);
            }
            mu.mtx.unlock();
        }
    }
    //update v via FTRL
    for(unordered_map<string, ftrl_model_unit*>::iterator iter = theta.begin(); iter != theta.end(); ++iter)
    {
        ftrl_model_unit& mu = *(iter->second);
        for(int f = 0; f < pModel->factor_num; ++f)
        {
            mu.mtx.lock();
            double& vif = mu.vi[f];
            double& v_nif = mu.v_ni[f];
            double& v_zif = mu.v_zi[f];
            if(v_nif > 0)
            {
                if(force_v_sparse && 0.0 == mu.wi)
                {
                    vif = 0.0;
                }
                else if(fabs(v_zif) <= v_l1)
                {
                    vif = 0.0;
                }
                else
                {
                    vif = (-1) *
                        (1 / (v_l2 + (v_beta + sqrt(v_nif)) / v_alpha)) *
                        (v_zif - utils::sgn(v_zif) * v_l1);
                }
            }
            mu.mtx.unlock();
        }
    }
    vector<double> sum1(pModel->factor_num);
    double p1 = pModel->predict(sample1.x, theta, sum1);
    vector<double> sum2(pModel->factor_num);
    double p2 = pModel->predict(sample2.x, theta, sum2);
    double lambda12 = -1 / (1 + exp(p1 - p2)) * deltaNDCG;
    //update w_n, w_z
    if(k1)
    {
        for(unordered_map<string, ftrl_model_unit*>::iterator iter = theta.begin(); iter != theta.end(); ++iter)
        {
            ftrl_model_unit& mu = *(iter->second);
            const string& index = iter->first;
            unordered_map<string, double>::iterator iter_x1 = sample1.x.find(index);
            double xi1 = (iter_x1 == sample1.x.end()) ? 0 : iter_x1->second;
            unordered_map<string, double>::iterator iter_x2 = sample2.x.find(index);
            double xi2 = (iter_x2 == sample2.x.end()) ? 0 : iter_x2->second;
            double xi = xi1 - xi2;
            if(xi != 0)
            {
                mu.mtx.lock();
                double w_gi = lambda12 * xi;
                double w_si = 1 / w_alpha * (sqrt(mu.w_ni + w_gi * w_gi) - sqrt(mu.w_ni));
                mu.w_zi += w_gi - w_si * mu.wi;
                mu.w_ni += w_gi * w_gi;
                mu.mtx.unlock();
            }
        }
    }
    //update v_n, v_z
    for(unordered_map<string, ftrl_model_unit*>::iterator iter = theta.begin(); iter != theta.end(); ++iter)
    {
        ftrl_model_unit& mu = *(iter->second);
        const string& index = iter->first;
        unordered_map<string, double>::iterator iter_x1 = sample1.x.find(index);
        double xi1 = (iter_x1 == sample1.x.end()) ? 0 : iter_x1->second;
        unordered_map<string, double>::iterator iter_x2 = sample2.x.find(index);
        double xi2 = (iter_x2 == sample2.x.end()) ? 0 : iter_x2->second;
        for(int f = 0; f < pModel->factor_num; ++f)
        {
            mu.mtx.lock();
            double& vif = mu.vi[f];
            double& v_nif = mu.v_ni[f];
            double& v_zif = mu.v_zi[f];
            double v_gif = lambda12 * ((sum1[f] * xi1 - vif * xi1 * xi1) - (sum2[f] * xi2 - vif * xi2 * xi2));
            double v_sif = 1 / v_alpha * (sqrt(v_nif + v_gif * v_gif) - sqrt(v_nif));
            v_zif += v_gif - v_sif * vif;
            v_nif += v_gif * v_gif;
            //有的特征在整个训练集中只出现一次，这里还需要对vif做一次处理
            if(force_v_sparse && v_nif > 0 && 0.0 == mu.wi)
            {
                vif = 0.0;
            }
            mu.mtx.unlock();
        }
    }
}

// Assembly:
pushq %rbp
pushq %r15
pushq %r14
pushq %r13
pushq %r12
pushq %rbx
subq $0x128, %rsp            # imm = 0x128
movsd %xmm0, 0x70(%rsp)
movq %rdx, %r12
movq %rsi, %r15
movq %rdi, %rbx
leaq 0xc8(%rsp), %rax
movq %rax, -0x30(%rax)
movq $0x1, -0x28(%rax)
xorpd %xmm0, %xmm0
movupd %xmm0, -0x20(%rax)
movl $0x3f800000, -0x10(%rax) # imm = 0x3F800000
movupd %xmm0, -0x8(%rax)
leaq 0x28(%rsi), %rax
movq %rax, 0x68(%rsp)
addq $0x38, %r15
leaq 0x108(%rsp), %rbp
movq (%r15), %r15
testq %r15, %r15
je 0x4625
leaq 0x8(%r15), %r14
leaq 0x98(%rsp), %rdi
movq %r14, %rsi
callq 0x74a4
testq %rax, %rax
jne 0x45c4
movq 0x8(%rbx), %r13
movq %rbp, %rdi
movq %r14, %rsi
callq 0x3320
movq %r13, %rdi
movq %rbp, %r13
movq %rbp, %rsi
callq 0x3706
movq %rax, %rbp
leaq 0x98(%rsp), %rdi
movq %r14, %rsi
callq 0x78c2
movq %rbp, (%rax)
movq %r13, %rbp
movq %r13, %rdi
callq 0x3468
jmp 0x45c4
leaq 0x28(%r12), %rax
movq %rax, 0x60(%rsp)
addq $0x38, %r12
leaq 0x98(%rsp), %r13
leaq 0xe8(%rsp), %rbp
movq (%r12), %r12
testq %r12, %r12
je 0x4696
leaq 0x8(%r12), %r14
movq %r13, %rdi
movq %r14, %rsi
callq 0x74a4
testq %rax, %rax
jne 0x4643
movq 0x8(%rbx), %r15
movq %rbp, %rdi
movq %r14, %rsi
callq 0x3320
movq %r15, %rdi
movq %rbp, %rsi
callq 0x3706
movq %rax, %r15
movq %r13, %rdi
movq %r14, %rsi
callq 0x78c2
movq %r15, (%rax)
movq %rbp, %rdi
callq 0x3468
jmp 0x4643
movq 0xa8(%rsp), %r14
cmpb $0x0, 0x50(%rbx)
je 0x47d0
testq %r14, %r14
je 0x47c8
movq 0x28(%r14), %r13
leaq 0x60(%r13), %r12
movq %r12, %rdi
callq 0x597c
movsd 0x10(%r13), %xmm0
andpd 0x6961(%rip), %xmm0     # 0xb030
movsd 0x20(%rbx), %xmm1
xorpd %xmm2, %xmm2
ucomisd %xmm0, %xmm1
jae 0x47b2
cmpb $0x1, 0x51(%rbx)
jne 0x4720
movsd 0x8(%r13), %xmm0
ucomisd 0x692a(%rip), %xmm0     # 0xb020
jbe 0x4720
movsd (%r13), %xmm0
ucomisd 0x691a(%rip), %xmm0     # 0xb020
jne 0x4720
jp 0x4720
movq 0x8(%rbx), %rax
movsd 0x40(%rax), %xmm1
movsd 0x48(%rax), %xmm0
movq %r13, %rdi
callq 0x5e98
movsd 0x18(%rbx), %xmm0
movsd %xmm0, 0x20(%rsp)
movsd 0x28(%rbx), %xmm0
movsd %xmm0, 0x10(%rsp)
movsd 0x8(%r13), %xmm0
ucomisd 0x68dc(%rip), %xmm0     # 0xb020
jb 0x474c
sqrtsd %xmm0, %xmm0
jmp 0x4751
callq 0x31f0
movsd %xmm0, 0x8(%rsp)
movsd 0x10(%rbx), %xmm0
movsd %xmm0, 0x30(%rsp)
movsd 0x10(%r13), %xmm0
movsd %xmm0, 0x50(%rsp)
callq 0xa258
movsd 0x20(%rsp), %xmm0
addsd 0x8(%rsp), %xmm0
divsd 0x30(%rsp), %xmm0
movsd 0x10(%rsp), %xmm1
addsd %xmm0, %xmm1
movsd 0x6881(%rip), %xmm0     # 0xb018
divsd %xmm1, %xmm0
cvtsi2sd %eax, %xmm1
mulsd 0x20(%rbx), %xmm1
movsd 0x50(%rsp), %xmm2
subsd %xmm1, %xmm2
mulsd %xmm0, %xmm2
movsd %xmm2, (%r13)
movq %r12, %rdi
callq 0x31d0
movq (%r14), %r14
jmp 0x46a8
movq 0xa8(%rsp), %r14
movq %rbx, 0x10(%rsp)
testq %r14, %r14
je 0x492d
movq 0x28(%r14), %r15
leaq 0x60(%r15), %r12
xorl %r13d, %r13d
movq %r14, 0x48(%rsp)
movq 0x8(%rbx), %rax
movslq 0x38(%rax), %rax
cmpq %rax, %r13
jge 0x4925
movq %r12, %rdi
callq 0x597c
movq 0x30(%r15), %rax
movsd (%rax,%r13,8), %xmm0
ucomisd 0x6807(%rip), %xmm0     # 0xb020
jbe 0x4915
movq %rbx, %rax
movq 0x18(%r15), %rbx
movq 0x48(%r15), %rbp
cmpb $0x1, 0x51(%rax)
jne 0x4845
movsd (%r15), %xmm1
xorpd %xmm3, %xmm3
ucomisd %xmm3, %xmm1
jne 0x4845
jnp 0x490a
movsd (%rbp,%r13,8), %xmm1
andpd 0x67dc(%rip), %xmm1     # 0xb030
movq 0x10(%rsp), %rax
movsd 0x40(%rax), %xmm2
xorpd %xmm3, %xmm3
ucomisd %xmm1, %xmm2
jae 0x490a
movq 0x10(%rsp), %rax
movsd 0x38(%rax), %xmm1
movsd %xmm1, 0x8(%rsp)
movsd 0x48(%rax), %xmm1
movsd %xmm1, 0x50(%rsp)
ucomisd 0x6791(%rip), %xmm0     # 0xb020
jb 0x4897
sqrtsd %xmm0, %xmm0
jmp 0x489c
callq 0x31f0
movsd %xmm0, 0x30(%rsp)
movq 0x10(%rsp), %r14
movsd 0x30(%r14), %xmm0
movsd %xmm0, 0x78(%rsp)
movsd (%rbp,%r13,8), %xmm0
movsd %xmm0, 0x20(%rsp)
callq 0xa258
movsd 0x8(%rsp), %xmm0
addsd 0x30(%rsp), %xmm0
divsd 0x78(%rsp), %xmm0
movsd 0x50(%rsp), %xmm1
addsd %xmm0, %xmm1
movsd 0x672f(%rip), %xmm0     # 0xb018
divsd %xmm1, %xmm0
cvtsi2sd %eax, %xmm1
mulsd 0x40(%r14), %xmm1
movsd 0x20(%rsp), %xmm3
subsd %xmm1, %xmm3
mulsd %xmm0, %xmm3
movq 0x48(%rsp), %r14
movsd %xmm3, (%rbx,%r13,8)
movq 0x10(%rsp), %rbx
movq %r12, %rdi
callq 0x31d0
incq %r13
jmp 0x47ee
movq (%r14), %r14
jmp 0x47d5
movq 0x8(%rbx), %rax
movslq 0x38(%rax), %rsi
leaq 0xd0(%rsp), %rdi
leaq 0x80(%rsp), %rdx
callq 0x5ef4
movq 0x8(%rbx), %rdi
leaq 0x98(%rsp), %rdx
leaq 0xd0(%rsp), %rcx
movq 0x68(%rsp), %rsi
callq 0x37be
movsd %xmm0, 0x20(%rsp)
movq 0x8(%rbx), %rax
movslq 0x38(%rax), %rsi
leaq 0x80(%rsp), %rdi
leaq 0x1f(%rsp), %rdx
callq 0x5ef4
movq 0x8(%rbx), %rdi
leaq 0x98(%rsp), %rdx
leaq 0x80(%rsp), %rcx
movq 0x60(%rsp), %rsi
callq 0x37be
movsd 0x20(%rsp), %xmm1
subsd %xmm0, %xmm1
movapd %xmm1, %xmm0
callq 0x3430
addsd 0x664f(%rip), %xmm0     # 0xb010
movsd 0x664f(%rip), %xmm1     # 0xb018
divsd %xmm0, %xmm1
mulsd 0x70(%rsp), %xmm1
movsd %xmm1, 0x48(%rsp)
movq 0xa8(%rsp), %r15
cmpb $0x1, 0x50(%rbx)
jne 0x4d00
testq %r15, %r15
je 0x4b2d
leaq 0x8(%r15), %r14
movq 0x28(%r15), %r13
movq 0x68(%rsp), %rdi
movq %r14, %rsi
callq 0x737a
testq %rax, %rax
je 0x4a15
movsd 0x28(%rax), %xmm0
jmp 0x4a19
xorpd %xmm0, %xmm0
movsd %xmm0, 0x20(%rsp)
movq 0x60(%rsp), %rdi
movq %r14, %rsi
callq 0x737a
testq %rax, %rax
je 0x4a38
movsd 0x28(%rax), %xmm0
jmp 0x4a3c
xorpd %xmm0, %xmm0
movsd 0x20(%rsp), %xmm1
subsd %xmm0, %xmm1
movsd %xmm1, 0x20(%rsp)
ucomisd 0x65cc(%rip), %xmm1     # 0xb020
jne 0x4a5c
jnp 0x4b25
leaq 0x60(%r13), %r12
movq %r12, %rdi
callq 0x597c
movsd 0x20(%rsp), %xmm0
mulsd 0x48(%rsp), %xmm0
movsd 0x10(%rbx), %xmm1
movsd %xmm1, 0x8(%rsp)
movsd %xmm0, 0x20(%rsp)
movapd %xmm0, %xmm1
mulsd %xmm0, %xmm1
movsd 0x8(%r13), %xmm0
movsd %xmm1, 0x50(%rsp)
addsd %xmm1, %xmm0
ucomisd 0x657b(%rip), %xmm0     # 0xb020
jb 0x4aad
sqrtsd %xmm0, %xmm1
jmp 0x4ab6
callq 0x31f0
movapd %xmm0, %xmm1
movsd 0x8(%r13), %xmm0
ucomisd 0x655c(%rip), %xmm0     # 0xb020
jb 0x4acc
sqrtsd %xmm0, %xmm0
jmp 0x4add
movsd %xmm1, 0x30(%rsp)
callq 0x31f0
movsd 0x30(%rsp), %xmm1
subsd %xmm0, %xmm1
movsd 0x652f(%rip), %xmm0     # 0xb018
divsd 0x8(%rsp), %xmm0
mulsd %xmm1, %xmm0
mulsd (%r13), %xmm0
addsd 0x20(%rsp), %xmm0
addsd 0x10(%r13), %xmm0
movsd %xmm0, 0x10(%r13)
movsd 0x50(%rsp), %xmm0
addsd 0x8(%r13), %xmm0
movsd %xmm0, 0x8(%r13)
movq %r12, %rdi
callq 0x31d0
movq (%r15), %r15
jmp 0x49eb
movq 0xa8(%rsp), %r15
jmp 0x4d00
leaq 0x8(%r15), %r14
movq 0x28(%r15), %r13
movq 0x68(%rsp), %rdi
movq %r14, %rsi
callq 0x737a
testq %rax, %rax
je 0x4b5b
movsd 0x28(%rax), %xmm0
jmp 0x4b5f
xorpd %xmm0, %xmm0
movapd %xmm0, 0x30(%rsp)
movq 0x60(%rsp), %rdi
movq %r14, %rsi
callq 0x737a
testq %rax, %rax
je 0x4b7e
movsd 0x28(%rax), %xmm0
jmp 0x4b82
xorpd %xmm0, %xmm0
movapd 0x30(%rsp), %xmm1
unpcklpd %xmm0, %xmm1            # xmm1 = xmm1[0],xmm0[0]
movapd %xmm1, 0x30(%rsp)
leaq 0x60(%r13), %r12
xorl %r14d, %r14d
movq %r15, 0x78(%rsp)
movq 0x8(%rbx), %rax
movslq 0x38(%rax), %rax
cmpq %rax, %r14
jge 0x4cfd
movq %r12, %rdi
callq 0x597c
movq 0x18(%r13), %rbp
movq 0x30(%r13), %r15
movq %rbx, %rdx
movq 0x48(%r13), %rbx
movq 0xd0(%rsp), %rax
movq 0x80(%rsp), %rcx
movsd (%rbp,%r14,8), %xmm0
unpcklpd %xmm0, %xmm0            # xmm0 = xmm0[0,0]
movapd 0x30(%rsp), %xmm1
mulpd %xmm1, %xmm0
mulpd %xmm1, %xmm0
movsd (%rax,%r14,8), %xmm2
movhpd (%rcx,%r14,8), %xmm2    # xmm2 = xmm2[0],mem[0]
mulpd %xmm1, %xmm2
subpd %xmm0, %xmm2
movapd %xmm2, %xmm0
unpckhpd %xmm2, %xmm0            # xmm0 = xmm0[1],xmm2[1]
subsd %xmm0, %xmm2
mulsd 0x48(%rsp), %xmm2
movsd 0x30(%rdx), %xmm0
movsd %xmm0, 0x8(%rsp)
movapd %xmm2, 0x20(%rsp)
mulsd %xmm2, %xmm2
movsd (%r15,%r14,8), %xmm0
movapd %xmm2, 0x50(%rsp)
addsd %xmm2, %xmm0
ucomisd 0x63de(%rip), %xmm0     # 0xb020
jb 0x4c4a
sqrtsd %xmm0, %xmm1
jmp 0x4c53
callq 0x31f0
movapd %xmm0, %xmm1
movsd (%r15,%r14,8), %xmm0
ucomisd 0x63bf(%rip), %xmm0     # 0xb020
jb 0x4c69
sqrtsd %xmm0, %xmm0
jmp 0x4c7a
movsd %xmm1, 0x70(%rsp)
callq 0x31f0
movsd 0x70(%rsp), %xmm1
subsd %xmm0, %xmm1
movsd 0x6392(%rip), %xmm0     # 0xb018
divsd 0x8(%rsp), %xmm0
mulsd %xmm1, %xmm0
mulsd (%rbp,%r14,8), %xmm0
addsd 0x20(%rsp), %xmm0
addsd (%rbx,%r14,8), %xmm0
movsd %xmm0, (%rbx,%r14,8)
movapd 0x50(%rsp), %xmm0
addsd (%r15,%r14,8), %xmm0
movsd %xmm0, (%r15,%r14,8)
movq 0x10(%rsp), %rbx
cmpb $0x1, 0x51(%rbx)
jne 0x4ce8
ucomisd 0x6352(%rip), %xmm0     # 0xb020
jbe 0x4ce8
movsd (%r13), %xmm0
ucomisd 0x6342(%rip), %xmm0     # 0xb020
jne 0x4ce8
jp 0x4ce8
andq $0x0, (%rbp,%r14,8)
movq %r12, %rdi
callq 0x31d0
incq %r14
movq 0x78(%rsp), %r15
jmp 0x4b9e
movq (%r15), %r15
testq %r15, %r15
jne 0x4b3a
leaq 0x80(%rsp), %rdi
callq 0x7496
leaq 0xd0(%rsp), %rdi
callq 0x7496
leaq 0x98(%rsp), %rdi
callq 0x6d16
addq $0x128, %rsp            # imm = 0x128
popq %rbx
popq %r12
popq %r13
popq %r14
popq %r15
popq %rbp
retq
jmp 0x4d82
jmp 0x4d46
movq %rax, %rbx
jmp 0x4d92
jmp 0x4da1
jmp 0x4d82
jmp 0x4da1
jmp 0x4da1
jmp 0x4d82
jmp 0x4d82
movq %rax, %rbx
leaq 0xe8(%rsp), %rdi
jmp 0x4d6f
movq %rax, %rbx
leaq 0x108(%rsp), %rdi
callq 0x3468
jmp 0x4da4
jmp 0x4da1
jmp 0x4d82
jmp 0x4d82
jmp 0x4da1
jmp 0x4da1
jmp 0x4da1
movq %rax, %rbx
leaq 0x80(%rsp), %rdi
callq 0x7496
leaq 0xd0(%rsp), %rdi
callq 0x7496
jmp 0x4da4
movq %rax, %rbx
leaq 0x98(%rsp), %rdi
callq 0x6d16
movq %rbx, %rdi
callq 0x3410
nop