public Network(LossFunctionType lossFunctionType, Optimizer optimizer, int numberOfClasses) { _numberOfClasses = numberOfClasses; _lossFunction = LossFunctionFactory.Produce(lossFunctionType); _layers = new List <IFullLayer>(); _optimizer = optimizer; }
public static LossFunction Produce(LossFunctionType type) { switch (type) { case LossFunctionType.CrossEntropy: return(new CrossEntropyLoss()); default: throw new Exception("Cost function is not supported"); } }