Несколько ярлыков с TensorFlow dnnRegression не работает

Question

У меня есть простая оценка dnnRegression, когда я хочу предсказать одну метку, она работает как положено, но когда я хочу предсказать несколько меток, она не работает для меня.Я знаю о label_dimension = 3, но у меня это не работает, не знаю, что я делаю не так.

Я пытался удалить учебную папку, играя с формами меток, но ничего не работаетЯ знаю, что target должен быть объектом серии, но не может заставить его работать с несколькими метками.

    def get_input_fn(data_set, data_target, num_epochs=None, n_batch=128, shuffle=False):
        return tf.estimator.inputs.pandas_input_fn(
            x=data_set,
            y=data_target,
            batch_size=n_batch,
            num_epochs=num_epochs,
            shuffle=shuffle
        )

    _LABEL = [
        'i1',
        'i2',
        'i3'
    ]


    train_set = pd.DataFrame(data=d1, columns=k1, dtype=float)
    train_target = pd.DataFrame(data=d2, columns=k1, dtype=float)


    #tt = pd.Series(data=train_target.iloc[:], index=_LABEL)
    #train_target = tt

    features = list(train_set.keys())

    feature_cols = [tf.feature_column.numeric_column(k) for k in features]

    estimator = tf.estimator.DNNRegressor(
        feature_columns=feature_cols,
        hidden_units=[1024, 512, 256],
        label_dimension=3,
        optimizer=tf.train.ProximalAdagradOptimizer(
            learning_rate=0.1,
            l1_regularization_strength=0.001
        ),
        model_dir="train"
    )

    estimator.train(
        input_fn=get_input_fn(data_set=train_set, data_target=train_target, num_epochs=None, n_batch=128, shuffle=False),
        steps=1000
    )

Я просто хотел бы получить прогнозы для нескольких меток, i1, i2 и i3.