Guest User

Untitled

a guest
Apr 20th, 2018
73
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.92 KB | None | 0 0
  1. k = 4;
  2. resource = ResourceObject["MNIST"];
  3. trainingData = ResourceData[resource, "TrainingData"];
  4. trainingSubset = Select[trainingData, Last[#] <= k &];
  5. testData = ResourceData[resource, "TestData"];
  6. testSubset = Select[testData, Last[#] <= k &];
  7. RandomSample[trainingSubset, 8]
  8. trainingImages = Keys[trainingSubset];
  9. meanImage = Image[Mean@Map[ImageData, trainingImages]]
  10.  
  11. net = NetGraph[{FlattenLayer[], 50, Ramp, 784, Tanh, ReshapeLayer[{1, 28, 28}],
  12. MeanSquaredLossLayer[]}, {1 -> 2 -> 3 -> 4 -> 5 -> 6 -> NetPort["Output"],
  13. 6 -> NetPort[7, "Input"], NetPort["Input"] -> NetPort[7, "Target"]},
  14. "Input" -> NetEncoder[{"Image", {28, 28}, "Grayscale", "MeanImage" -> meanImage}],
  15. "Output" -> NetDecoder[{"Image", "Grayscale"}]]
  16.  
  17. trained = NetTrain[net, <|"Input" -> trainingImages|>, "Loss"];
  18.  
  19. cm = ClassifierMeasurements[trained, testData]
  20.  
  21. ClassifierMeasurements: This neural network cannot be converted to a ClassifierFunction.
Add Comment
Please, Sign In to add comment