Will Koehrsen WillKoehrsen

🌆

building

Senior Software Engineer @gridstatus

WillKoehrsen / training_with_early_stopping.py

Created November 26, 2018 18:45

	# Early stopping details
	n_epochs_stop = 5
	min_val_loss = np.Inf
	epochs_no_improve = 0

	# Main loop
	for epoch in range(n_epochs):
	# Initialize validation loss for epoch
	val_loss = 0

WillKoehrsen / pytorch_training.py

Created November 26, 2018 16:14

	for epoch in range(n_epochs):
	for data, targets in trainloader:
	# Generate predictions
	out = model(data)
	# Calculate loss
	loss = criterion(out, targets)
	# Backpropagation
	loss.backward()
	# Update model parameters
	optimizer.step()

WillKoehrsen / dataloaders.py

Created November 26, 2018 15:47

	from torchvision import datasets
	from torch.utils.data import DataLoader

	# Datasets from folders
	data = {
	'train':
	datasets.ImageFolder(root=traindir, transform=image_transforms['train']),
	'valid':
	datasets.ImageFolder(root=validdir, transform=image_transforms['valid']),
	}

WillKoehrsen / pytorch_transforms.py

Created November 26, 2018 15:44

	from torchvision import transforms

	# Image transformations
	image_transforms = {
	# Train uses data augmentation
	'train':
	transforms.Compose([
	transforms.RandomResizedCrop(size=256, scale=(0.8, 1.0)),
	transforms.RandomRotation(degrees=15),
	transforms.ColorJitter(),

WillKoehrsen / random.html

Created November 18, 2018 02:20

	<!DOCTYPE html>
	<html>

	<header>
	<title>Random Starting Abstract
	</title>

	<link rel="stylesheet" href="/static/css/main.css">
	<link rel="shortcut icon" href="/static/images/lstm.ico">
	<ul>

WillKoehrsen / load_keras_model.py

Last active February 15, 2019 23:50

	from keras.models import load_model
	import tensorflow as tf

	def load_keras_model():
	"""Load in the pre-trained model"""
	global model
	model = load_model('../models/train-embeddings-rnn.h5')
	# Required for model to work
	global graph
	graph = tf.get_default_graph()

WillKoehrsen / serve_home.py

Created November 18, 2018 02:15

	from flask import request
	# User defined utility functions
	from utils import generate_random_start, generate_from_seed

	# Home page
	@app.route("/", methods=['GET', 'POST'])
	def home():
	"""Home page of app with form"""

	# Create form

WillKoehrsen / rnn_index.html

Created November 18, 2018 02:10

	<!DOCTYPE html>
	<html>

	<head>
	<title>RNN Patent Writing</title>
	<link rel="stylesheet" href="/static/css/main.css">
	<link rel="shortcut icon" href="/static/images/lstm.ico">

	</head>

WillKoehrsen / form_serve.py

Last active November 18, 2018 02:09

	from flask import render_template

	# Home page
	@app.route("/", methods=['GET', 'POST'])
	def home():
	"""Home page of app with form"""
	# Create form
	form = ReusableForm(request.form)

	# Send template information to index.html

WillKoehrsen / form.py

Created November 18, 2018 02:03

	from wtforms import (Form, TextField, validators, SubmitField,
	DecimalField, IntegerField)

	class ReusableForm(Form):
	"""User entry form for entering specifics for generation"""
	# Starting seed
	seed = TextField("Enter a seed string or 'random':", validators=[
	validators.InputRequired()])
	# Diversity of predictions
	diversity = DecimalField('Enter diversity:', default=0.8,