from rest_framework import serializers
from .models import MalwarePrediction

class MalwarePredictionSerializer(serializers.ModelSerializer):
    class Meta:
        model = MalwarePrediction
        fields = ['id', 'process_name', 'process_class', 'probability_of_malware','model_type']
