-
Notifications
You must be signed in to change notification settings - Fork 33
/
Copy pathopts.py
37 lines (29 loc) · 868 Bytes
/
opts.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# Copyright © 2016 Peng Liu <liupeng@imscv.com>
#
# Distributed under terms of the GNU GPL3 license.
"""
This file implement a Opt class for storing some hyper-parameter of DL.
"""
class DLOption(object):
"""Docstring for DLOption. """
def __init__(self, epoches, learning_rate, batchsize, momentum, penaltyL2,
dropoutProb):
"""TODO: to be defined1.
:epoches: TODO
:learning_rate: TODO
:batchsize: TODO
:momentum: TODO
:penaltyL2: TODO
:dropout: TODO
:dropoutProb: TODO
"""
self._epoches = epoches
self._learning_rate = learning_rate
self._batchsize = batchsize
self._momentum = momentum
self._penaltyL2 = penaltyL2
self._dropoutProb = dropoutProb