opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-02 02:35:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Changed thumbnail of nn1

Browse Source
This commit is contained in:
Grant Sanderson
2019-05-24 15:04:45 -07:00
parent c579c7d99f
commit b9822db5bf
2 changed files with 41 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
old_projects/nn/network.py
View File

@ -20,7 +20,7 @@ import random
import numpy as np import numpy as np
from PIL import Image from PIL import Image
from nn.mnist_loader import load_data_wrapper from nn.mnist_loader import load_data_wrapper
from utils.space_ops import get_norm # from utils.space_ops import get_norm
NN_DIRECTORY = os.path.dirname(os.path.realpath(__file__)) NN_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
# PRETRAINED_DATA_FILE = os.path.join(NN_DIRECTORY, "pretrained_weights_and_biases_80") # PRETRAINED_DATA_FILE = os.path.join(NN_DIRECTORY, "pretrained_weights_and_biases_80")

47
old_projects/nn/part1.py
View File

@ -2964,7 +2964,7 @@ class BiasForInactiviyWords(Scene):
self.play(Write(words)) self.play(Write(words))
self.wait(3) self.wait(3)
class ContinualEdgeUpdate(ContinualAnimation): class ContinualEdgeUpdate(VGroup):
CONFIG = { CONFIG = {
"max_stroke_width" : 3, "max_stroke_width" : 3,
"stroke_width_exp" : 7, "stroke_width_exp" : 7,
@ -2972,7 +2972,8 @@ class ContinualEdgeUpdate(ContinualAnimation):
"colors" : [GREEN, GREEN, GREEN, RED], "colors" : [GREEN, GREEN, GREEN, RED],
} }
def __init__(self, network_mob, **kwargs): def __init__(self, network_mob, **kwargs):
digest_config(self, kwargs) VGroup.__init__(self, **kwargs)
self.internal_time = 0
n_cycles = self.n_cycles n_cycles = self.n_cycles
edges = VGroup(*it.chain(*network_mob.edge_groups)) edges = VGroup(*it.chain(*network_mob.edge_groups))
self.move_to_targets = [] self.move_to_targets = []
@ -2990,11 +2991,14 @@ class ContinualEdgeUpdate(ContinualAnimation):
edge.generate_target() edge.generate_target()
edge.target.set_stroke(edge.colors[0], edge.widths[0]) edge.target.set_stroke(edge.colors[0], edge.widths[0])
self.move_to_targets.append(MoveToTarget(edge)) edge.become(edge.target)
self.move_to_targets.append(edge)
self.edges = edges self.edges = edges
ContinualAnimation.__init__(self, edges, **kwargs) self.add(edges)
self.add_updater(lambda m, dt: self.update_edges(dt))
def update_mobject(self, dt): def update_edges(self, dt):
self.internal_time += dt
if self.internal_time < 1: if self.internal_time < 1:
alpha = smooth(self.internal_time) alpha = smooth(self.internal_time)
for move_to_target in self.move_to_targets: for move_to_target in self.move_to_targets:
@ -4559,8 +4563,9 @@ class ShowAmplify(PiCreatureScene):
class Thumbnail(NetworkScene): class Thumbnail(NetworkScene):
CONFIG = { CONFIG = {
"network_mob_config" : { "network_mob_config" : {
'neuron_stroke_color' : WHITE 'neuron_stroke_color' : WHITE,
} 'layer_to_layer_buff': 1.25,
},
} }
def construct(self): def construct(self):
network_mob = self.network_mob network_mob = self.network_mob
@ -4568,6 +4573,23 @@ class Thumbnail(NetworkScene):
for layer in network_mob.layers: for layer in network_mob.layers:
layer.neurons.set_stroke(width = 5) layer.neurons.set_stroke(width = 5)
network_mob.set_height(5)
network_mob.to_edge(DOWN)
network_mob.to_edge(LEFT, buff=1)
subtitle = TextMobject(
"From the\\\\",
"ground up\\\\",
)
# subtitle.arrange(
# DOWN,
# buff=0.25,
# aligned_edge=LEFT,
# )
subtitle.set_color(YELLOW)
subtitle.set_height(2.75)
subtitle.next_to(network_mob, RIGHT, buff=MED_LARGE_BUFF)
edge_update = ContinualEdgeUpdate( edge_update = ContinualEdgeUpdate(
network_mob, network_mob,
max_stroke_width = 10, max_stroke_width = 10,
@ -4576,7 +4598,18 @@ class Thumbnail(NetworkScene):
edge_update.internal_time = 3 edge_update.internal_time = 3
edge_update.update(0) edge_update.update(0)
for mob in network_mob.family_members_with_points():
if mob.get_stroke_width() < 2:
mob.set_stroke(width=2)
title = TextMobject("Neural Networks")
title.scale(3)
title.to_edge(UP)
self.add(network_mob) self.add(network_mob)
self.add(subtitle)
self.add(title)