Spaces:

cvachet
/

iris_classification_lambda

Sleeping

Clement Vachet

Use list of features as direct input

f4e0e4a almost 2 years ago

644 Bytes

from classification.classifier import Classifier
import json


if __name__ == "__main__":
    cls = Classifier()

    # Training
    cls.train_and_save()

    # Testing
    data = {
        "features": [[6.5, 3.0, 5.8, 2.2],[6.1, 2.8, 4.7, 1.2]]
    }
    features = data["features"]
    results = cls.load_and_test(features)
    print("results:", results)

    # Response similar to REST API call
    response = {
        'statusCode': 200,
        'body': json.dumps({
            'predictions': results["predictions"],
            'probabilities': results["probabilities"]
        })
    }
    print("Example REST API response: ", response)

	from classification.classifier import Classifier
	import json


	if __name__ == "__main__":
	cls = Classifier()

	# Training
	cls.train_and_save()

	# Testing
	data = {
	"features": [[6.5, 3.0, 5.8, 2.2],[6.1, 2.8, 4.7, 1.2]]
	}
	features = data["features"]
	results = cls.load_and_test(features)
	print("results:", results)

	# Response similar to REST API call
	response = {
	'statusCode': 200,
	'body': json.dumps({
	'predictions': results["predictions"],
	'probabilities': results["probabilities"]
	})
	}
	print("Example REST API response: ", response)