python - 使用 tabpy 预测性别计数的结果

Question

我有一个由日期时间索引和男性和女性计数组成的年度数据，我在表格计算字段中创建了一个模型，下面是代码

SCRIPT_REAL(
"
import pandas as pd
import numpy as np
from statsmodels.tsa.arima.model import ARIMA
from random import random

period = _arg3[0]+1

model = ARIMA(_arg2[:-period], order=(2,1,2))
model_fit= model.fit()
future = model_fit.predict(1,len(_arg2), typ='levels')

print(_arg2[:-period])

return future.tolist()

",ATTR([Date]),[Provider Count],[Forecast Year])

我得到了计数的预测，但是当我尝试使用性别过滤器时，我确实得到了结果，但是男性预测和女性预测的总和不等于总计数的预测。当我使用过滤器 Gender='M' 时，我怀疑正在发生的事情，然后只传递男性计数数据并预测结果，女性和总计数相同。我想要的是使用总计数来预测模型，当我使用过滤器 gender='M' 时，总计数的预测结果应该用于显示男性和女性计数。