SnowFlash383935's picture
Update README.md
8bd97f9 verified
metadata
license: gpl-3.0
metrics:
  - accuracy
pipeline_tag: tabular-classification
tags:
  - finance
library_name: transformers

Модель для соревнования DigitalEdu с использованием трансформеров. 9 эпох.

Параметры

  • Пол
  • Имеет ли фото
  • Привязан ли телефон
  • Количество подписчиков
  • Семейное положение
  • Главное в жизни
  • Главное в людях

Пример

from transformers import AutoModel
model = AutoModel.from_pretrained("SnowFlash383935/DigitalEduTransformers", trust_remote_code=True)
model([
    [1, 1, 1, 150, 2, 0, 0],
    [0, 1, 1, 294, 0, 0, 0],
])

Пример вывода: [False, True]