Simple Neural Network Model
This is a simple neural network model designed for text classification tasks. The model uses a basic feedforward neural network architecture, trained for binary classification.
Model Details
- Architecture: A simple feedforward neural network with one hidden layer.
- Number of classes: 2 (binary classification)
- Input size: 768 (or as per your dataset)
Usage
To use this model with Hugging Face Transformers:
from transformers import AutoModel, AutoTokenizer
# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("your-username/simple-nn-model")
model = AutoModel.from_pretrained("your-username/simple-nn-model")
# Tokenize inputs
inputs = tokenizer("Sample input text", return_tensors="pt")
# Get model predictions
outputs = model(**inputs)
- Downloads last month
- 10
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.